diff --git a/core/modules/migrate/tests/src/Unit/process/MigrationTest.php b/core/modules/migrate/tests/src/Unit/process/MigrationTest.php
index cade905..06d02bc 100644
--- a/core/modules/migrate/tests/src/Unit/process/MigrationTest.php
+++ b/core/modules/migrate/tests/src/Unit/process/MigrationTest.php
@@ -108,7 +108,7 @@ public function testTransformWithStubbing() {
    */
   protected function getMigration() {
     $id_map = $this->prophesize(MigrateIdMapInterface::class);
-    $id_map->lookupDestinationId([1])->willReturn(NULL);
+    $id_map->lookupDestinationIds([1])->willReturn(NULL);
     $id_map->saveIdMapping(Argument::any(), Argument::any(), MigrateIdMapInterface::STATUS_NEEDS_UPDATE)->willReturn(NULL);
 
     $migration = $this->prophesize(MigrationInterface::class);
@@ -150,7 +150,7 @@ public function testSuccessfulLookup($source_id_values, $destination_id_values,
     $this->migration_plugin->id()->willReturn(uniqid());
 
     $id_map = $this->prophesize(MigrateIdMapInterface::class);
-    $id_map->lookupDestinationId($source_id_values)->willReturn($destination_id_values);
+    $id_map->lookupDestinationIds($source_id_values)->willReturn([$destination_id_values]);
     $this->migration_plugin->getIdMap()->willReturn($id_map->reveal());
 
     $this->migration_plugin_manager->createInstances(['foobaz'])