diff --git a/src/Viserio/Middleware/Dispatcher.php b/src/Viserio/Middleware/Dispatcher.php
index 55d35d9dc..5d175708c 100644
--- a/src/Viserio/Middleware/Dispatcher.php
+++ b/src/Viserio/Middleware/Dispatcher.php
@@ -104,9 +104,9 @@ private function nextFrame()
     /**
      *  Check if middleware is aware of Interop\Container\ContainerInterface.
      *
-     * @param MiddlewareContract $middleware
+     * @param \Viserio\Contracts\Middleware\Middleware $middleware
      *
-     * @return MiddlewareContract
+     * @return \Viserio\Contracts\Middleware\Middleware
      */
     private function isContainerAware($middleware): MiddlewareContract
     {