diff --git a/src/dynapyt/post_run.py b/src/dynapyt/post_run.py
index 016018c..fdb7feb 100644
--- a/src/dynapyt/post_run.py
+++ b/src/dynapyt/post_run.py
@@ -4,8 +4,9 @@
 from .utils.runtimeUtils import gather_coverage, gather_output
 
 
-def post_run(coverage_dir: str, output_dir: str):
-    gather_output(Path(output_dir))
+def post_run(coverage_dir: str = "", output_dir: str = ""):
+    if len(output_dir) > 0:
+        gather_output(Path(output_dir))
     if len(coverage_dir) > 0:
         gather_coverage(Path(coverage_dir))
 
diff --git a/src/dynapyt/runtime.py b/src/dynapyt/runtime.py
index ea02dda..6071ee3 100644
--- a/src/dynapyt/runtime.py
+++ b/src/dynapyt/runtime.py
@@ -52,7 +52,7 @@ def __init__(self):
     def __del__(self):
         self.end_execution()
 
-    def end_execution(self):
+    def end_execution(self, *args):
         if self.end_execution_called:
             return
         self.end_execution_called = True