1 files changed, 4 insertions, 3 deletions
diff --git a/tests/Router/Runner/ControllerTest.php b/tests/Router/Runner/ControllerTest.php
index 26d0578..84acef4 100644
--- a/tests/Router/Runner/ControllerTest.php
+++ b/tests/Router/Runner/ControllerTest.php
@@ -7,8 +7,8 @@ use Jasny\Router\Runner;
 use Jasny\Router\ControllerFactory;
 use Psr\Http\Message\ServerRequestInterface;
 use Psr\Http\Message\ResponseInterface;
-
-use Jasny\Router\TestHelpers;
+use Jasny\TestHelper;
+use Jasny\Router\MockResponse;
 
 /**
  * @covers Jasny\Router\Runner\Controller
@@ -17,7 +17,8 @@ use Jasny\Router\TestHelpers;
  */
 class ControllerTest extends \PHPUnit_Framework_TestCase
 {
-    use TestHelpers;
+    use TestHelper;
+    use MockResponse;
     
     public function testInvoke()
     {