diff --git a/modules/salesforce_pull/tests/src/Unit/PullBaseTest.php b/modules/salesforce_pull/tests/src/Unit/PullBaseTest.php
index d93cf363d52c59797a2d8df8890f5238f1d41e91..f8f3e5561a3a60a4c578f5dd826ea4641372c665 100644
--- a/modules/salesforce_pull/tests/src/Unit/PullBaseTest.php
+++ b/modules/salesforce_pull/tests/src/Unit/PullBaseTest.php
@@ -308,16 +308,14 @@ class PullBaseTest extends UnitTestCase {
       ->willReturn($this->entityDefinition);
     $this->etm = $prophecy->reveal();
 
-    $this->pullWorker = $this->getMockBuilder(PullBase::CLASS)
-      ->setConstructorArgs([
-        $this->etm,
-        $this->sfapi,
-        $this->ed,
-        [],
-        'cron_salesforce_pull',
-        ['cron' => ['time' => 180]],
-      ])
-      ->getMockForAbstractClass();
+    $this->pullWorker = new CronPull(
+      $this->etm,
+      $this->sfapi,
+      $this->ed,
+      [],
+      'cron_salesforce_pull',
+      ['cron' => ['time' => 180]],
+    );
 
     $sobject = new SObject([
       'id' => $this->salesforce_id,
diff --git a/modules/salesforce_pull/tests/src/Unit/QueueHandlerTest.php b/modules/salesforce_pull/tests/src/Unit/QueueHandlerTest.php
index e83c65b77a11d2eba3220ec0a7480b4c2adb4f54..0dd81a238c5afcedf268ac65a856226ecd7a98c2 100644
--- a/modules/salesforce_pull/tests/src/Unit/QueueHandlerTest.php
+++ b/modules/salesforce_pull/tests/src/Unit/QueueHandlerTest.php
@@ -156,16 +156,14 @@ class QueueHandlerTest extends UnitTestCase {
 
     $this->time = $this->getMockBuilder(TimeInterface::CLASS)->getMock();
 
-    $this->qh = $this->getMockBuilder(QueueHandler::CLASS)
-      ->setConstructorArgs([
+    $this->qh = new QueueHandler(
         $this->sfapi,
         $this->etm,
         $this->queue_factory,
         $this->configFactory,
         $this->ed,
-        $this->time,
-      ])
-      ->getMock();
+        $this->time
+      );
   }
 
   /**
@@ -198,16 +196,14 @@ class QueueHandlerTest extends UnitTestCase {
     $prophecy->get(Argument::any())->willReturn($this->queue);
     $this->queue_factory = $prophecy->reveal();
 
-    $this->qh = $this->getMockBuilder(QueueHandler::CLASS)
-      ->setConstructorArgs([
+    $this->qh = new QueueHandler(
         $this->sfapi,
         $this->etm,
         $this->queue_factory,
         $this->configFactory,
         $this->ed,
-        $this->time,
-      ])
-      ->getMock();
+        $this->time
+    );
     $result = $this->qh->getUpdatedRecords();
     $this->assertFalse($result);
   }