diff --git a/packages/native-core-macro/src/lib.rs b/packages/native-core-macro/src/lib.rs
index c8b58efba..5ab2c8388 100644
--- a/packages/native-core-macro/src/lib.rs
+++ b/packages/native-core-macro/src/lib.rs
@@ -68,7 +68,7 @@ fn impl_derive_macro(ast: &syn::DeriveInput) -> TokenStream {
         .map(|field| &field.ty);
 
     let gen = quote! {
-        impl dioxus_native_core::State for #type_name {
+        impl dioxus_native_core::State<#custom_type> for #type_name {
             fn create_passes() -> Box<[dioxus_native_core::TypeErasedPass<Self>]> {
                 Box::new([
                     #(