diff --git a/tests/unit/Codeception/Module/RestTest.php b/tests/unit/Codeception/Module/RestTest.php
index cd1c4e7..114800a 100644
--- a/tests/unit/Codeception/Module/RestTest.php
+++ b/tests/unit/Codeception/Module/RestTest.php
@@ -697,18 +697,16 @@ public function testRestExecute(string $configUrl, string $requestUrl, string $e
         $connectionModule
             ->expects($this->once())
             ->method('_request')
-            ->will(
-                $this->returnCallback(function($method,
-                    $uri,
-                    $parameters,
-                    $files,
-                    $server,
-                    $content
-                ) use ($expectedFullUrl) {
-                    Assert::assertSame($expectedFullUrl, $uri);
-                    return '';
-                })
-            );
+            ->willReturnCallback(function($method,
+                $uri,
+                $parameters,
+                $files,
+                $server,
+                $content
+            ) use ($expectedFullUrl) {
+                Assert::assertSame($expectedFullUrl, $uri);
+                return '';
+            });
 
         $config = ['url' => $configUrl];