diff --git a/immediate_stats/src/bevy/butler.rs b/immediate_stats/src/bevy/butler.rs
index 83dfb3f..8bc8117 100644
--- a/immediate_stats/src/bevy/butler.rs
+++ b/immediate_stats/src/bevy/butler.rs
@@ -1,14 +1,12 @@
 #[cfg(test)]
 mod tests {
+    use crate as immediate_stats;
     use crate::StatContainer;
     use crate::stat::Stat;
     use bevy_app::App;
     use bevy_butler::*;
     use bevy_ecs::prelude::*;
     use bevy_reflect::prelude::*;
-    use immediate_stats_macros::StatContainer;
-
-    use crate as immediate_stats;
 
     #[butler_plugin]
     struct MyPlugin;
diff --git a/immediate_stats/src/lib.rs b/immediate_stats/src/lib.rs
index 6e63e4e..cdd0d45 100644
--- a/immediate_stats/src/lib.rs
+++ b/immediate_stats/src/lib.rs
@@ -2,6 +2,7 @@
 pub mod bevy;
 pub mod stat;
 
+pub use bevy_app::prelude::PreUpdate;
 pub use immediate_stats_macros::StatContainer;
 
 /// Types that contain stats that need to be reset.
diff --git a/immediate_stats_macros/src/bevy/butler.rs b/immediate_stats_macros/src/bevy/butler.rs
index dd36517..2660e22 100644
--- a/immediate_stats_macros/src/bevy/butler.rs
+++ b/immediate_stats_macros/src/bevy/butler.rs
@@ -37,7 +37,7 @@ pub(crate) fn register_butler_systems(tree: &DeriveInput) -> TokenStream {
             let input = input.parse::<proc_macro2::Ident>().expect("An identifier.");
 
             systems.push(quote! {
-                #[bevy_butler::add_system(generics = <#struct_name>, plugin = #input, schedule = bevy_app::PreUpdate)]
+                #[bevy_butler::add_system(generics = <#struct_name>, plugin = #input, schedule = immediate_stats::PreUpdate)]
                 #use_system
             });