diff --git a/tests/phpunit/MigrateTest.php b/tests/phpunit/MigrateTest.php
index 985e63f..f51477d 100644
--- a/tests/phpunit/MigrateTest.php
+++ b/tests/phpunit/MigrateTest.php
@@ -713,6 +713,11 @@ public function testMigrateShouldHandleEncryptionApiErrorResponse(): void
                 ];
             });
 
+        $sourceClientMock
+            ->method('generateId')
+            ->willReturn('123')
+        ;
+
         /** @var JobRunner $sourceJobRunnerMock */
         /** @var JobRunner $destJobRunnerMock */
         $migrate = new Migrate(