diff --git a/diracx-testing/src/diracx/testing/mock_osdb.py b/diracx-testing/src/diracx/testing/mock_osdb.py
index 455665e4..ae4007c2 100644
--- a/diracx-testing/src/diracx/testing/mock_osdb.py
+++ b/diracx-testing/src/diracx/testing/mock_osdb.py
@@ -101,7 +101,7 @@ async def upsert(self, doc_id, document) -> None:
             await self._sql_db.conn.execute(stmt)
 
     async def bulk_insert(self, index_name: str, docs: list[dict[str, Any]]) -> None:
-        async with self:
+        async with self._sql_db:
             rows = []
             for doc in docs:
                 # don't use doc_id column explicitly. This ensures that doc_id is unique.
@@ -169,7 +169,7 @@ async def search(
         return results
 
     async def delete(self, query: list[dict[str, Any]]) -> None:
-        async with self:
+        async with self._sql_db:
             stmt = delete(self._table)
             stmt = sql_utils.apply_search_filters(
                 self._table.columns.__getitem__, stmt, query