diff --git a/models.py b/models.py
index 5673525..f832932 100644
--- a/models.py
+++ b/models.py
@@ -2,7 +2,8 @@ from __future__ import annotations
 
 from datetime import datetime
 
-from eloquent import Model, QueryBuilder  # type: ignore
+from eloquent import Model  # type: ignore
+from eloquent.query import QueryBuilder  # type: ignore
 
 from .config import DB