diff --git a/crates/bevy_ecs/src/schedule/state.rs b/crates/bevy_ecs/src/schedule/state.rs index 763198aa0f..27141d05fc 100644 --- a/crates/bevy_ecs/src/schedule/state.rs +++ b/crates/bevy_ecs/src/schedule/state.rs @@ -1,6 +1,7 @@ use std::fmt::Debug; use std::hash::Hash; use std::mem; +use std::ops::Deref; use crate as bevy_ecs; use crate::change_detection::DetectChangesMut; @@ -95,6 +96,14 @@ impl PartialEq for State { } } +impl Deref for State { + type Target = S; + + fn deref(&self) -> &Self::Target { + self.get() + } +} + /// The next state of [`State`]. /// /// To queue a transition, just set the contained value to `Some(next_state)`.