diff --git a/main.py b/main.py
index ad7cf296e25c576ed16341811d6edd8f8ecd4798..dedee957f94dd6cd4a20341633a8e30ee494726c 100644
--- a/main.py
+++ b/main.py
@@ -89,7 +89,7 @@ def sample_func(f):
   print(f"SAMPLING SETTINGS:\n\n {sampling_setting}\n\n")
 
   print("\n\nSTART SAMPLING\n\n")
-  globals()[meta_setting["sampling_function"]](model=framework,device=device,sample_all=sample_all, **sampling_setting,)
+  globals()[meta_setting["sampling_function"]](model=framework,device=device, **sampling_setting,)
   print("\n\nFINISHED SAMPLING\n\n")