diff --git a/core/modules/migrate/tests/src/Unit/MigrateExecutableTest.php b/core/modules/migrate/tests/src/Unit/MigrateExecutableTest.php
index e65e629..851a1a8 100644
--- a/core/modules/migrate/tests/src/Unit/MigrateExecutableTest.php
+++ b/core/modules/migrate/tests/src/Unit/MigrateExecutableTest.php
@@ -437,10 +437,7 @@ public function testProcessRowPipelineException() {
     $plugin->getPluginId()->willReturn('plugin_id');
     $plugin = $plugin->reveal();
     $plugins['destination_id'] = [$plugin, $plugin];
-    $this->migration->expects($this->once())
-      ->method('getProcessPlugins')
-      ->with(NULL)
-      ->will($this->returnValue($plugins));
+    $this->migration->method('getProcessPlugin')->willReturn($plugins);
 
     $this->executable->processRow($row);
   }