diff --git a/firedantic/_async/model.py b/firedantic/_async/model.py
index 84251df..a84b34f 100644
--- a/firedantic/_async/model.py
+++ b/firedantic/_async/model.py
@@ -144,7 +144,7 @@ def get_document_id(self):
     @classmethod
     async def find(  # pylint: disable=too-many-arguments
         cls: Type[TAsyncBareModel],
-        filter_: Optional[Dict[str, Union[str, dict]]] = None,
+        filter_: Optional[Dict[str, Union[str, int, float, bool, dict]]] = None,
         order_by: Optional[_OrderBy] = None,
         limit: Optional[int] = None,
         offset: Optional[int] = None,
@@ -224,7 +224,7 @@ def _add_filter(
     @classmethod
     async def find_one(
         cls: Type[TAsyncBareModel],
-        filter_: Optional[Dict[str, Union[str, dict]]] = None,
+        filter_: Optional[Dict[str, Union[str, int, float, bool, dict]]] = None,
         order_by: Optional[_OrderBy] = None,
         transaction: Optional[AsyncTransaction] = None,
     ) -> TAsyncBareModel:
diff --git a/firedantic/_sync/model.py b/firedantic/_sync/model.py
index 78ec372..28adc1b 100644
--- a/firedantic/_sync/model.py
+++ b/firedantic/_sync/model.py
@@ -144,7 +144,7 @@ def get_document_id(self):
     @classmethod
     def find(  # pylint: disable=too-many-arguments
         cls: Type[TBareModel],
-        filter_: Optional[Dict[str, Union[str, dict]]] = None,
+        filter_: Optional[Dict[str, Union[str, int, float, bool, dict]]] = None,
         order_by: Optional[_OrderBy] = None,
         limit: Optional[int] = None,
         offset: Optional[int] = None,
@@ -224,7 +224,7 @@ def _add_filter(
     @classmethod
     def find_one(
         cls: Type[TBareModel],
-        filter_: Optional[Dict[str, Union[str, dict]]] = None,
+        filter_: Optional[Dict[str, Union[str, int, float, bool, dict]]] = None,
         order_by: Optional[_OrderBy] = None,
         transaction: Optional[Transaction] = None,
     ) -> TBareModel: