diff --git a/tests/unit/lib/classes/MigrationTest.php b/tests/unit/lib/classes/MigrationTest.php
index 1f0bc98c56b320f52af4499d411cb1c6f9c34f06..0aabbcd42c6824c0ac9ea06acb42f4bfc5de4d47 100644
--- a/tests/unit/lib/classes/MigrationTest.php
+++ b/tests/unit/lib/classes/MigrationTest.php
@@ -41,6 +41,11 @@ class MigrationTest extends \Codeception\Test\Unit
         {
             private $versions = [0];
 
+            public function getDomain()
+            {
+                return 'test';
+            }
+
             public function getBranch()
             {
                 return 0;
@@ -81,7 +86,7 @@ class MigrationTest extends \Codeception\Test\Unit
         $migrator->migrateTo(2);
 
         $relevant = $migrator->relevantMigrations(null);
-        $this->assertSame(1, count($relevant));
+        $this->assertSame(2, count($relevant));
     }
 
     public function testMigrationUp()