diff --git a/src/exploration.rs b/src/exploration.rs
index 9838c86..a1443e3 100644
--- a/src/exploration.rs
+++ b/src/exploration.rs
@@ -421,7 +421,8 @@ where
         if !config.states.is_empty() {
             let states = config.states;
             for state in states {
-                app.add_systems(OnExit(state), cleanup);
+                app.configure_set(Update, Exploration.run_if(in_state(state.clone())))
+                    .add_systems(OnExit(state), cleanup);
             }
         }
     }