diff --git a/tests/unit/test_serp.py b/tests/unit/test_serp.py
index b1bd739..359b36b 100644
--- a/tests/unit/test_serp.py
+++ b/tests/unit/test_serp.py
@@ -248,7 +248,13 @@ class TestAsyncEndpoints:
             "videos": 50,
             "audio": 10
         })
-        mock_db.session = MagicMock()
+        # Mock session as async context manager
+        mock_session = MagicMock()
+        mock_session.execute = AsyncMock(return_value=MagicMock(scalar=MagicMock(return_value=0)))
+        mock_db.session = MagicMock(return_value=MagicMock(
+            __aenter__=AsyncMock(return_value=mock_session),
+            __aexit__=AsyncMock(return_value=None)
+        ))
 
         transport = ASGITransport(app=app)
         async with AsyncClient(transport=transport, base_url="http://test") as client: