diff --git a/luisa_compute_derive_impl/src/lib.rs b/luisa_compute_derive_impl/src/lib.rs
index 3791fd2..c40e21c 100644
--- a/luisa_compute_derive_impl/src/lib.rs
+++ b/luisa_compute_derive_impl/src/lib.rs
@@ -141,7 +141,8 @@ impl Compiler {
                     #(self.#field_names.encode(encoder);)*
                 }
             }
-            impl #impl_generics #runtime_path::AsKernelArg<#name #ty_generics> for #name #ty_generics #where_clause {
+            impl #impl_generics #runtime_path::AsKernelArg for #name #ty_generics #where_clause {
+                type Output = #name #ty_generics;
             }
         )
     }