1 files changed, 2 insertions, 6 deletions
diff --git a/tests/Unit/Middleware/LegacyMiddlewareTest.php b/tests/Unit/Middleware/LegacyMiddlewareTest.php
index 34e60b60..ed9a5a74 100644
--- a/tests/Unit/Middleware/LegacyMiddlewareTest.php
+++ b/tests/Unit/Middleware/LegacyMiddlewareTest.php
@@ -46,10 +46,11 @@ class LegacyMiddlewareTest extends TestCase
                 ['title2', 'content2']
             );
 
-        $middleware->expects($this->exactly(2))
+        $middleware->expects($this->exactly(3))
             ->method('renderPage')
             ->withConsecutive(
                 ['user_worklog', 'title', 'content'],
+                ['404', 'Page not found'],
                 ['login', 'title2', 'content2']
             )
             ->willReturn($response);
@@ -73,11 +74,6 @@ class LegacyMiddlewareTest extends TestCase
                 '/'
             );
 
-        $handler->expects($this->once())
-            ->method('handle')
-            ->with($request)
-            ->willReturn($response);
-
         $middleware->process($request, $handler);
         $middleware->process($request, $handler);
         $middleware->process($request, $handler);