diff --git a/src/replit/database/__init__.py b/src/replit/database/__init__.py
index e64393c..b8b222b 100644
--- a/src/replit/database/__init__.py
+++ b/src/replit/database/__init__.py
@@ -27,6 +27,7 @@ class JSONKey:
 
     def __init__(
             self,
+            db: Any,
             key: str,
             dtype: JSON_TYPE,
             get_default: Callable = None,