diff --git a/src/backend/langflow/services/database/models/__init__.py b/src/backend/langflow/services/database/models/__init__.py
index 01e81e277..3cc4231a3 100644
--- a/src/backend/langflow/services/database/models/__init__.py
+++ b/src/backend/langflow/services/database/models/__init__.py
@@ -1,6 +1,5 @@
 from .flow import Flow
 from .user import User
-from .token import Token
 from .api_key import ApiKey
 
-__all__ = ["Flow", "User", "Token", "ApiKey"]
+__all__ = ["Flow", "User", "ApiKey"]
diff --git a/src/backend/langflow/services/database/models/token/__init__.py b/src/backend/langflow/services/database/models/token/__init__.py
deleted file mode 100644
index 9b9fa397d..000000000
--- a/src/backend/langflow/services/database/models/token/__init__.py
+++ /dev/null
@@ -1,5 +0,0 @@
-from .token import Token
-
-__all__ = [
-    "Token",
-]
diff --git a/src/backend/langflow/services/database/models/token/token.py b/src/backend/langflow/services/database/models/token/token.py
deleted file mode 100644
index 68c70f07f..000000000
--- a/src/backend/langflow/services/database/models/token/token.py
+++ /dev/null
@@ -1,7 +0,0 @@
-from pydantic import BaseModel
-
-
-class Token(BaseModel):
-    access_token: str
-    refresh_token: str
-    token_type: str
diff --git a/tests/test_custom_component.py b/tests/test_custom_component.py
index 4dc8c9f1a..e75dc0e5b 100644
--- a/tests/test_custom_component.py
+++ b/tests/test_custom_component.py
@@ -473,15 +473,16 @@ def test_build_config_no_code():
 
 
 @pytest.fixture
-def component():
+def component(client, active_user):
     return CustomComponent(
+        user_id=active_user.id,
         field_config={
             "fields": {
                 "llm": {"type": "str"},
                 "url": {"type": "str"},
                 "year": {"type": "int"},
             }
-        }
+        },
     )