diff --git a/models.py b/models.py
index 2ec8270..85e3808 100644
--- a/models.py
+++ b/models.py
@@ -3,6 +3,7 @@ from __future__ import annotations
 from datetime import datetime
 
 from eloquent import Model  # type: ignore
+from eloquent.orm import belongs_to
 from eloquent.query import QueryBuilder  # type: ignore
 
 from .config import DB