diff --git a/tests/framework/console/controllers/ServeControllerTest.php b/tests/framework/console/controllers/ServeControllerTest.php
index 463251e33b2..cbf95e392cf 100644
--- a/tests/framework/console/controllers/ServeControllerTest.php
+++ b/tests/framework/console/controllers/ServeControllerTest.php
@@ -29,7 +29,7 @@ public function testAddressTaken()
         $docroot = __DIR__ . '/stub';
 
         /** @var ServeController $serveController */
-        $serveController = $this->getMockBuilder(ServeControllerMock::class)
+        $serveController = $this->getMockBuilder('yii\console\controllers\ServeController')
             ->setConstructorArgs(['serve', Yii::$app])
             ->setMethods(['isAddressTaken'])
             ->getMock();