diff --git a/crates/ui-egui/src/lib.rs b/crates/ui-egui/src/lib.rs
index be9b7c939f..5da9751cfc 100644
--- a/crates/ui-egui/src/lib.rs
+++ b/crates/ui-egui/src/lib.rs
@@ -55,8 +55,6 @@ mod system_fonts;
 mod tab_context;
 pub mod theme;
 mod widgets;
-#[cfg(not(target_arch = "wasm32"))]
-pub mod window_state;
 
 #[cfg(all(test, not(target_arch = "wasm32")))]
 mod open_worker_tests;
@@ -111,6 +109,8 @@ pub use persist::last_session;
 pub use persist::last_view;
 use persist::recovery;
 pub use persist::units;
+#[cfg(not(target_arch = "wasm32"))]
+pub use persist::window_state;
 use platform::drag_pointer;
 #[cfg(not(target_arch = "wasm32"))]
 pub use platform::file_watch;
diff --git a/crates/ui-egui/src/persist.rs b/crates/ui-egui/src/persist.rs
index a3c1df3731..bb29efeff2 100644
--- a/crates/ui-egui/src/persist.rs
+++ b/crates/ui-egui/src/persist.rs
@@ -2,3 +2,5 @@ pub mod last_session;
 pub mod last_view;
 pub(super) mod recovery;
 pub mod units;
+#[cfg(not(target_arch = "wasm32"))]
+pub mod window_state;
diff --git a/crates/ui-egui/src/window_state.rs b/crates/ui-egui/src/persist/window_state.rs
similarity index 100%
rename from crates/ui-egui/src/window_state.rs
rename to crates/ui-egui/src/persist/window_state.rs