diff --git a/Connection.php b/Connection.php
index 971bb4f..cd7c5e3 100644
--- a/Connection.php
+++ b/Connection.php
@@ -14,21 +14,21 @@ namespace Database;
 use Database\Affair\BeginTransaction;
 use Database\Affair\Commit;
 use Database\Affair\Rollback;
-use PDO;
 use Database\Mysql\Schema;
 use Exception;
 use Kiri;
-use Kiri\Context;
 use Kiri\Abstracts\Component;
 use Kiri\Abstracts\Config;
+use Kiri\Di\ContainerInterface;
+use Kiri\Di\Context;
 use Kiri\Events\EventProvider;
 use Kiri\Exception\NotFindClassException;
+use Kiri\Pool\Connection as PoolConnection;
 use Kiri\Server\Events\OnWorkerExit;
+use PDO;
 use Psr\Container\ContainerExceptionInterface;
 use Psr\Container\NotFoundExceptionInterface;
 use ReflectionException;
-use Kiri\Pool\Connection as PoolConnection;
-use Kiri\Di\ContainerInterface;
 
 /**
  * Class Connection
diff --git a/Db.php b/Db.php
index 4aaa825..040611d 100644
--- a/Db.php
+++ b/Db.php
@@ -14,7 +14,7 @@ use Database\Affair\Rollback;
 use Database\Traits\QueryTrait;
 use Exception;
 use Kiri\Abstracts\Config;
-use Kiri\Context;
+use Kiri\Di\Context;
 use Kiri\Events\EventDispatch;
 use Kiri\Exception\ConfigException;
 use Psr\Container\ContainerExceptionInterface;
diff --git a/Relation.php b/Relation.php
index 0712510..21ea593 100644
--- a/Relation.php
+++ b/Relation.php
@@ -4,9 +4,8 @@ declare(strict_types=1);
 namespace Database;
 
 
-use Exception;
 use Kiri\Abstracts\Component;
-use Kiri\Context;
+use Kiri\Di\Context;
 
 /**
  * Class Relation