diff --git a/core/modules/migrate/tests/src/Unit/MemoryManagerTest.php b/core/modules/migrate/tests/src/Unit/MemoryManagerTest.php
index 9a049972d5..ecca18836f 100644
--- a/core/modules/migrate/tests/src/Unit/MemoryManagerTest.php
+++ b/core/modules/migrate/tests/src/Unit/MemoryManagerTest.php
@@ -3,6 +3,8 @@
 namespace Drupal\Tests\migrate\Unit;
 
 use Drupal\migrate\MemoryManager;
+use Drupal\migrate\MigrateMessageInterface;
+use Symfony\Contracts\EventDispatcher\EventDispatcherInterface;
 
 /**
  * Tests the memory manager.
@@ -30,8 +32,8 @@ class MemoryManagerTest extends MigrateTestCase {
    */
   protected function setUp(): void {
     parent::setUp();
-    $this->message = $this->createMock('Drupal\migrate\MigrateMessageInterface');
-    $this->eventDispatcher = $this->createMock('Symfony\Component\EventDispatcher\EventDispatcherInterface');
+    $this->message = $this->createMock(MigrateMessageInterface::class);
+    $this->eventDispatcher = $this->createMock(EventDispatcherInterface::class);
   }
 
   /**
diff --git a/core/modules/migrate/tests/src/Unit/MigrateExecutableTest.php b/core/modules/migrate/tests/src/Unit/MigrateExecutableTest.php
index dcb74d8b9a..969e47edff 100644
--- a/core/modules/migrate/tests/src/Unit/MigrateExecutableTest.php
+++ b/core/modules/migrate/tests/src/Unit/MigrateExecutableTest.php
@@ -460,6 +460,9 @@ public function testRollback(array $id_map_records, bool $rollback_called = TRUE
       ->method('getDestinationPlugin')
       ->willReturn($destination->reveal());
 
+    $this->memoryManager
+      ->method('ensureMemory')
+      ->willReturn(TRUE);
     $executable = new TestMigrateExecutable($migration, $this->message, $this->eventDispatcher, $this->memoryManager);
 
     $this->assertEquals($expected_result, $executable->rollback());