diff --git a/src/fiber_package/entity/submodels.py b/src/fiber_package/entity/submodels.py
index 3d56547..8aefb84 100644
--- a/src/fiber_package/entity/submodels.py
+++ b/src/fiber_package/entity/submodels.py
@@ -1,17 +1,20 @@
 from dataclasses import dataclass, field
 from datetime import date
 
-# always use @dataclass with slots=True for better performance
+from pydantic import BaseModel
 
-@dataclass(slots=True)
-class LoginInfo:
+
+
+# always use @dataclass with slots=True for better performance
+# unless if you have unions like LoginInfo and Login...
+
+class LoginInfo(BaseModel):
     username: str
     hashed_password: str
     refresh_tokens: list[str] = field(default_factory=list)
     disabled: bool = False
 
-@dataclass(slots=True)
-class LoginInfoResponse:
+class LoginInfoResponse(BaseModel):
     username: str
 
 @dataclass(slots=True)