diff --git a/python/cachebox/utils.py b/python/cachebox/utils.py
index f9d832e..b520d28 100644
--- a/python/cachebox/utils.py
+++ b/python/cachebox/utils.py
@@ -463,7 +463,7 @@ def cache_clear() -> None:
 
 
 def cached(
-    cache: typing.Union[BaseCacheImpl, dict, None],
+    cache: typing.Union[BaseCacheImpl, dict, typing.Callable[..., BaseCacheImpl], None],
     key_maker: typing.Callable[[tuple, dict], typing.Hashable] = make_key,
     clear_reuse: bool = False,
     callback: typing.Optional[typing.Callable[[int, typing.Any, typing.Any], typing.Any]] = None,