Update main.py

main.py

@@ -321,7 +321,7 @@ class BuildSustain:
                 if key in st.session_state:
                     st.session_state.pop(key, None)
         
-        # Clear components-specific states when leaving the Building Components page
+        # Clear component-specific states when leaving the Building Components page
         if previous_page == "Building Components":
             keys_to_clear = [
                 "walls_editor",
@@ -350,11 +350,11 @@ class BuildSustain:
         """
         # Check for any module-specific rerun requests
         for module_name, should_rerun in st.session_state.module_rerun_flags.items():
-            if should_rer How to play:
-            # Clear the flag and trigger rerun
-            st.session_state.module_rerun_flags[module_name] = False
-            st.rerun()
-            break  # Only handle one rerun at a time
+            if should_rerun:
+                # Clear the flag and trigger rerun
+                st.session_state.module_rerun_flags[module_name] = False
+                st.rerun()
+                break  # Only handle one rerun at a time
     
     def display_page(self, page: str):
         """