diff --git a/src/backend/langflow/services/database/models/__init__.py b/src/backend/langflow/services/database/models/__init__.py
index da47bc5fe..28d2b4af8 100644
--- a/src/backend/langflow/services/database/models/__init__.py
+++ b/src/backend/langflow/services/database/models/__init__.py
@@ -1,4 +1,5 @@
 from .flow import Flow
+from .user import User
+from .token import Token
 
-
-__all__ = ["Flow"]
+__all__ = ["Flow", "User", "Token"]