diff --git a/test/WhoopsErrorHandlerTest.php b/test/WhoopsErrorHandlerTest.php
index 15ad77ac..1a24e05f 100644
--- a/test/WhoopsErrorHandlerTest.php
+++ b/test/WhoopsErrorHandlerTest.php
@@ -45,12 +45,13 @@ public function testExceptionErrorPreparesPageHandlerAndInvokesWhoops()
     {
         $exception = new Exception('Boom!');
 
-        $whoops = $this->getWhoops();
-        $whoops->handleException($exception)->willReturn('Whoops content');
-
         $pageHandler = $this->getPrettyPageHandler();
         $pageHandler->addDataTable('Expressive Application Request', Argument::type('array'))->shouldBeCalled();
 
+        $whoops = $this->getWhoops();
+        $whoops->handleException($exception)->willReturn('Whoops content');
+        $whoops->pushHandler(Argument::type(PrettyPageHandler::class))->shouldBeCalled();
+
         $handler = new WhoopsErrorHandler($whoops->reveal(), $pageHandler->reveal());
 
         $stream    = $this->prophesize(StreamInterface::class);