diff --git a/watertap/ui/fsapi.py b/watertap/ui/fsapi.py
index 37065ba676..b6dffc8dc9 100644
--- a/watertap/ui/fsapi.py
+++ b/watertap/ui/fsapi.py
@@ -828,7 +828,7 @@ def add_action(self, action_name: str, action_func: Callable):
 
         # print(f'ADDING ACTION: {action_name}')
         # print(action_func)
-        def action_wrapper(**kwargs):
+        def action_wrapper(*args, **kwargs):
             if action_name == Actions.build:
                 # set new model object from return value of build action
                 action_result = action_func(**kwargs)