diff --git a/demo-project/app/Handlers/CustomExceptionHandler.php b/demo-project/app/Handlers/CustomExceptionHandler.php
index 019a1b2..d758ae7 100644
--- a/demo-project/app/Handlers/CustomExceptionHandler.php
+++ b/demo-project/app/Handlers/CustomExceptionHandler.php
@@ -7,7 +7,7 @@ use Pecee\SimpleRouter\RouterEntry;
 
 class CustomExceptionHandler implements IExceptionHandler {
 
-    public function handleError( Request $request, RouterEntry $router = null, \Exception $error) {
+    public function handleError( Request $request, RouterEntry &$route = null, \Exception $error) {
 
         // Return json errors if we encounter an error on /api.
         if(stripos($request->getUri(), '/api') !== false) {
diff --git a/demo-project/app/Middlewares/ApiVerification.php b/demo-project/app/Middlewares/ApiVerification.php
index 2f71561..e97f0f2 100644
--- a/demo-project/app/Middlewares/ApiVerification.php
+++ b/demo-project/app/Middlewares/ApiVerification.php
@@ -3,10 +3,11 @@ namespace Demo\Middlewares;
 
 use Pecee\Http\Middleware\IMiddleware;
 use Pecee\Http\Request;
+use Pecee\SimpleRouter\RouterEntry;
 
 class ApiVerification implements IMiddleware {
 
-    public function handle(Request &$request) {
+    public function handle(Request $request, RouterEntry &$route = null) {
 
         // Do authentication
         $request->authenticated = true;