diff --git a/crates/bevy_ecs/src/schedule/state.rs b/crates/bevy_ecs/src/schedule/state.rs
index d19c9861f3d86..5565868f3057d 100644
--- a/crates/bevy_ecs/src/schedule/state.rs
+++ b/crates/bevy_ecs/src/schedule/state.rs
@@ -319,7 +319,9 @@ where
     /// # }
     /// #
     /// fn change_state(mut state: ResMut<State<AppState>>) {
-    ///     state.set(AppState::Game).unwrap().end_frame();
+    ///     if let Ok(transition) = state.set(AppState::Game) {
+    ///         transition.end_frame();
+    ///     }
     /// }
     /// # change_state.system();
     /// ```