diff --git a/Tests/Command/SetupTransportsCommandTest.php b/Tests/Command/SetupTransportsCommandTest.php
index c31e71a..ab5f29b 100644
--- a/Tests/Command/SetupTransportsCommandTest.php
+++ b/Tests/Command/SetupTransportsCommandTest.php
@@ -101,9 +101,7 @@ public function testThrowsExceptionOnTransportSetup()
         $serviceLocator = $this->createMock(ServiceLocator::class);
         $serviceLocator->expects($this->exactly(1))
             ->method('get')
-            ->will($this->onConsecutiveCalls(
-                $amqpTransport
-            ));
+            ->willReturn($amqpTransport);
         $serviceLocator
             ->method('has')
             ->willReturn(true);