diff --git a/immediate_stats/src/bevy/butler.rs b/immediate_stats/src/bevy/butler.rs index a60537b..386f00b 100644 --- a/immediate_stats/src/bevy/butler.rs +++ b/immediate_stats/src/bevy/butler.rs @@ -8,6 +8,8 @@ mod tests { use bevy_reflect::prelude::*; use immediate_stats_macros::StatContainer; + use crate as immediate_stats; + #[butler_plugin] struct MyPlugin; diff --git a/immediate_stats_macros/src/bevy_butler.rs b/immediate_stats_macros/src/bevy_butler.rs index 3ccd5cf..59bff56 100644 --- a/immediate_stats_macros/src/bevy_butler.rs +++ b/immediate_stats_macros/src/bevy_butler.rs @@ -15,8 +15,10 @@ pub(super) fn register_butler_systems(tree: &DeriveInput) -> TokenStream { }; let use_system = match ident.to_string().as_str() { - "resource" => quote! { use crate::bevy::reset_resource_modifiers; }, - "stat_butler_component" => quote! { use crate::bevy::reset_component_modifiers; }, + "resource" => quote! { use immediate_stats::bevy::reset_resource_modifiers; }, + "stat_butler_component" => { + quote! { use immediate_stats::bevy::reset_component_modifiers; } + } _ => continue, };