diff --git a/demo/Model/Job.php b/demo/Model/Job.php
index 27e0a4e..f3cb9ca 100644
--- a/demo/Model/Job.php
+++ b/demo/Model/Job.php
@@ -5,11 +5,12 @@ namespace Fufle\Demo\Model;
 use Fufle\Demo\Query\IJobQuery;
 use Fufle\Demo\Query\JobQuery;
 use Fufle\ORM\Model as BaseModel;
+use Fufle\ORM\IQuery;
 
 class Job extends BaseModel {
     public int $id;
     public float $budget;
-    public static function SetQueryModel() {
+    public static function SetQueryModel(): IQuery {
         return new JobQuery( static::class );
     }