diff --git a/ray_provider/decorators/ray.py b/ray_provider/decorators/ray.py
index ee51872..f63aac2 100644
--- a/ray_provider/decorators/ray.py
+++ b/ray_provider/decorators/ray.py
@@ -160,7 +160,7 @@ def task(
         if config is None:
             config = {}
         elif callable(config):
-            config = config()
+            config = config(**kwargs)
         elif not isinstance(config, dict):
             raise TypeError("config must be either a callable, a dictionary, or None")