lang: Fix compilation error when an #[account] struct has generics (#…

…3148)
coral-xyz · Aug 3, 2024 · 4853cd1 · 4853cd1
1 parent dc6ac2d
commit 4853cd1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lang/attribute/account/src/lib.rs b/lang/attribute/account/src/lib.rs
@@ -100,7 +100,11 @@ pub fn account(
         );
         format!("{discriminator:?}").parse().unwrap()
     };
-    let disc = quote! { #account_name::DISCRIMINATOR };
+    let disc = if account_strct.generics.lt_token.is_some() {
+        quote! { #account_name::#type_gen::DISCRIMINATOR }
+    } else {
+        quote! { #account_name::DISCRIMINATOR }
+    };
 
     let owner_impl = {
         if namespace.is_empty() {