diff --git a/crates/hedz_reloaded/src/protocol/mod.rs b/crates/hedz_reloaded/src/protocol/mod.rs
index b3cb879..5363e25 100644
--- a/crates/hedz_reloaded/src/protocol/mod.rs
+++ b/crates/hedz_reloaded/src/protocol/mod.rs
@@ -163,8 +163,8 @@ impl NetworkEnv<'_> {
     }
 }
 
-pub fn is_server(state: Res<State<ClientState>>) -> bool {
-    matches!(**state, ClientState::Disconnected)
+pub fn is_server(env: NetworkEnv) -> bool {
+    env.is_server()
 }
 
 fn set_game_tick(on: On<SetGameTick>, mut tick: ResMut<GameTick>) {