diff --git a/src/plugin.rs b/src/plugin.rs
index 6b8421cc..b6312657 100644
--- a/src/plugin.rs
+++ b/src/plugin.rs
@@ -100,6 +100,11 @@ impl<A: Actionlike> Plugin for InputManagerPlugin<A> {
                 )
                 .add_systems(PostUpdate, release_on_input_map_removed::<A>);
 
+                app.add_systems(
+                    PreUpdate,
+                    update_action_state::<A>.in_set(InputManagerSystem::Update),
+                );
+
                 app.configure_set(
                     PreUpdate,
                     InputManagerSystem::Update
@@ -130,11 +135,6 @@ impl<A: Actionlike> Plugin for InputManagerPlugin<A> {
                         .after(InputSystem),
                 );
 
-                app.add_systems(
-                    PreUpdate,
-                    update_action_state::<A>.in_set(InputManagerSystem::Update),
-                );
-
                 #[cfg(feature = "ui")]
                 app.add_systems(
                     PreUpdate,