diff --git a/tests/test_login.py b/tests/test_login.py
index f505f4100..29a8a852e 100644
--- a/tests/test_login.py
+++ b/tests/test_login.py
@@ -1,17 +1,15 @@
-from langflow.services.database.utils import session_getter
-from langflow.services.getters import get_db_service
 import pytest
-from langflow.services.database.models.user import User
 from langflow.services.auth.utils import get_password_hash
+from langflow.services.database.models.user import User
+from langflow.services.database.utils import session_getter
+from langflow.services.deps import get_db_service
 
 
 @pytest.fixture
 def test_user():
     return User(
         username="testuser",
-        password=get_password_hash(
-            "testpassword"
-        ),  # Assuming password needs to be hashed
+        password=get_password_hash("testpassword"),  # Assuming password needs to be hashed
         is_active=True,
         is_superuser=False,
     )
@@ -23,17 +21,13 @@ def test_login_successful(client, test_user):
         session.add(test_user)
         session.commit()
 
-    response = client.post(
-        "api/v1/login", data={"username": "testuser", "password": "testpassword"}
-    )
+    response = client.post("api/v1/login", data={"username": "testuser", "password": "testpassword"})
     assert response.status_code == 200
     assert "access_token" in response.json()
 
 
 def test_login_unsuccessful_wrong_username(client):
-    response = client.post(
-        "api/v1/login", data={"username": "wrongusername", "password": "testpassword"}
-    )
+    response = client.post("api/v1/login", data={"username": "wrongusername", "password": "testpassword"})
     assert response.status_code == 401
     assert response.json()["detail"] == "Incorrect username or password"
 
@@ -43,8 +37,6 @@ def test_login_unsuccessful_wrong_password(client, test_user, session):
     session.add(test_user)
     session.commit()
 
-    response = client.post(
-        "api/v1/login", data={"username": "testuser", "password": "wrongpassword"}
-    )
+    response = client.post("api/v1/login", data={"username": "testuser", "password": "wrongpassword"})
     assert response.status_code == 401
     assert response.json()["detail"] == "Incorrect username or password"