diff --git a/src/Base/Controller.php b/src/Base/Controller.php
index b9233f0..0867ca0 100644
--- a/src/Base/Controller.php
+++ b/src/Base/Controller.php
@@ -5,7 +5,7 @@ namespace Kiri\Router\Base;
 
 
 use Kiri;
-use Kiri\Di\ContainerInterface;
+use Psr\Container\ContainerInterface;
 use Psr\Log\LoggerInterface;
 use Psr\Http\Message\RequestInterface;
 use Psr\Http\Message\ResponseInterface;
diff --git a/src/Server.php b/src/Server.php
index 556ee21..bfcd686 100644
--- a/src/Server.php
+++ b/src/Server.php
@@ -6,7 +6,7 @@ namespace Kiri\Router;
 
 use Exception;
 use Kiri;
-use Kiri\Di\ContainerInterface;
+use Psr\Container\ContainerInterface;
 use Kiri\Di\Context;
 use Kiri\Events\EventProvider;
 use Kiri\Router\Interface\ExceptionHandlerInterface;