diff --git a/main.py b/main.py
index c54b158..ee4ae0f 100644
--- a/main.py
+++ b/main.py
@@ -217,12 +217,12 @@ def automate_function(
 
     ifc.write(ifc_filename)
     print(f"\nIFC file written: {ifc_filename}")
-    # try:
-    #     automate_context.mark_run_success("Success! You can download the IF file below.")
-    #     automate_context.store_file_result(f"./{ifc_filename}")
-    # except Exception as e:
-    #     print(f"  ⚠️  Could not upload file result (network issue?): {e}")
-    #     automate_context.mark_run_failed(f"Something went wrong when storing file result. Exception detail: {e}") 
+    try:
+        automate_context.mark_run_success("Success! You can download the IF file below.")
+        automate_context.store_file_result(f"./{ifc_filename}")
+    except Exception as e:
+        print(f"  ⚠️  Could not upload file result (network issue?): {e}")
+        automate_context.mark_run_failed(f"Something went wrong when storing file result. Exception detail: {e}") 
 
     print(f"\n{'=' * 60}")
     print(f"  Export complete!")