diff --git a/core/modules/migrate/tests/src/Unit/MigrateSourceTest.php b/core/modules/migrate/tests/src/Unit/MigrateSourceTest.php
index f37f041..3e3d906 100644
--- a/core/modules/migrate/tests/src/Unit/MigrateSourceTest.php
+++ b/core/modules/migrate/tests/src/Unit/MigrateSourceTest.php
@@ -186,13 +186,14 @@ public function testCount() {
 
     $this->migrationConfiguration['id'] = 'test_migration';
     $migration = $this->getMigration();
-    $source = new StubSourceGeneratorPlugin([], '', [], $migration);
+    $module_handler = $this->getMock(ModuleHandlerInterface::class);
+    $source = new StubSourceGeneratorPlugin([], '', [], $migration, $module_handler, $cache);
 
     // Test the skipCount property's default value.
     $this->assertEquals(-1, $source->count());
 
     // Test the count value using a generator.
-    $source = new StubSourceGeneratorPlugin(['skip_count' => FALSE], '', [], $migration);
+    $source = new StubSourceGeneratorPlugin(['skip_count' => FALSE], '', [], $migration, $module_handler, $cache);
     $this->assertEquals(3, $source->count());
   }
 
@@ -421,7 +422,9 @@ public function testPrepareRowPrepareException() {
   public function testDefaultPropertiesValues() {
     $this->migrationConfiguration['id'] = 'test_migration';
     $migration = $this->getMigration();
-    $source = new StubSourceGeneratorPlugin([], '', [], $migration);
+    $module_handler = $this->getMock(ModuleHandlerInterface::class);
+    $cache = $this->getMock(CacheBackendInterface::class);
+    $source = new StubSourceGeneratorPlugin([], '', [], $migration, $module_handler, $cache);
 
     // Test the default value of the skipCount Value;
     $this->assertTrue($source->getSkipCount());