diff --git a/Base/Model.php b/Base/Model.php
index e7173db..f4f6038 100644
--- a/Base/Model.php
+++ b/Base/Model.php
@@ -17,6 +17,7 @@ use ArrayAccess;
 use Database\ActiveQuery;
 use Database\Collection;
 use Database\Connection;
+use Database\ISqlBuilder;
 use Database\ModelInterface;
 use Database\Mysql\Columns;
 use Database\Relation;
diff --git a/Traits/HasBase.php b/Traits/HasBase.php
index 5166e9c..ddffd9a 100644
--- a/Traits/HasBase.php
+++ b/Traits/HasBase.php
@@ -46,16 +46,19 @@ abstract class HasBase implements \Database\Traits\Relation
 	public function __construct(public string $name)
 	{
 	}
-	
+
 	/**
 	 * @param $name
 	 * @param $arguments
 	 * @return static
+	 * @throws \ReflectionException
 	 */
 	public function __call($name, $arguments)
 	{
 		if ($name !== 'get') {
-			return di(Relation::class)->getQuery($this->name)->$name(...$arguments);
+			$relation = Kiri::getDi()->get(Relation::class);
+			$relation->getQuery($this->name)->$name(...$arguments);
+			return $this;
 		} else {
 			return $this->get();
 		}