diff --git a/src/database/core.py b/src/database/core.py
index e67f6f6..46b553d 100644
--- a/src/database/core.py
+++ b/src/database/core.py
@@ -1,7 +1,7 @@
 # src/database.py
 import functools
 import re
-from contextlib import contextmanager
+from contextlib import asynccontextmanager, contextmanager
 from typing import Annotated, Any, AsyncGenerator
 
 from fastapi import Depends