diff --git a/Abstracts/BaseProcess.php b/Abstracts/BaseProcess.php
index 3e8676c..06326e6 100644
--- a/Abstracts/BaseProcess.php
+++ b/Abstracts/BaseProcess.php
@@ -3,11 +3,9 @@
 namespace Kiri\Server\Abstracts;
 
 
-use Kiri\Context;
-use Kiri\Server\Broadcast\OnBroadcastInterface;
+use Kiri\Di\Context;
 use Kiri\Server\Contract\OnProcessInterface;
 use Swoole\Coroutine;
-use Swoole\Process;
 
 /**
  *
diff --git a/Abstracts/DoWhile.php b/Abstracts/DoWhile.php
index 6f0a33c..243fc44 100644
--- a/Abstracts/DoWhile.php
+++ b/Abstracts/DoWhile.php
@@ -2,7 +2,7 @@
 
 namespace Kiri\Server\Abstracts;
 
-use Kiri\Context;
+use Kiri\Di\Context;
 
 class DoWhile
 {