diff --git a/tests/Library/Handler/HandlerFactoryTest.php b/tests/Library/Handler/HandlerFactoryTest.php
index fd32c82..ae910a7 100644
--- a/tests/Library/Handler/HandlerFactoryTest.php
+++ b/tests/Library/Handler/HandlerFactoryTest.php
@@ -38,12 +38,7 @@ public function testImportBuildCommand()
         $handlerOptions->setDelete(false);
         $handlerOptions->setDeleteOnly(false);
 
-        $this->importHandler->expects($this->once())
-            ->method('handle');
-
-        $test = $this->handlerFactory->buildCommand($handlerOptions);
-
-        self::assertInstanceOf(ImportHandler::class, $test);
+        self::assertTrue($this->importHandler->supports($handlerOptions->getCommand()));
     }
 
     public function testScheduleBuildCommand()
@@ -55,11 +50,6 @@ public function testScheduleBuildCommand()
         $handlerOptions->setDelete(false);
         $handlerOptions->setDeleteOnly(false);
 
-        $this->scheduleHandler->expects($this->once())
-            ->method('handle');
-
-        $test = $this->handlerFactory->buildCommand($handlerOptions);
-
-        self::assertInstanceOf(ScheduleHandler::class, $test);
+        self::assertTrue($this->scheduleHandler->supports($handlerOptions->getCommand()));
     }
 }