diff --git a/packages/syft/src/syft/service/service.py b/packages/syft/src/syft/service/service.py index 55f2c1f4b5d..cda115cb8b4 100644 --- a/packages/syft/src/syft/service/service.py +++ b/packages/syft/src/syft/service/service.py @@ -289,7 +289,7 @@ def reconstruct_args_kwargs( else: raise Exception(f"Missing {param_key} not in kwargs.") - if "context": + if "context" in kwargs: final_kwargs["context"] = kwargs["context"] return (args, final_kwargs)