diff --git a/tools/src/main.rs b/tools/src/main.rs
index 1154b8f..e5f5c50 100644
--- a/tools/src/main.rs
+++ b/tools/src/main.rs
@@ -198,7 +198,7 @@ fn parse_struct(item_struct: &ItemStruct) -> Option<RustDecl> {
         methods: Vec::new(),
         return_type: None,
         parameters: Vec::new(),
-        docstring: extract_docstring(&item_fn.attrs),
+        docstring: extract_docstring(&item_struct.attrs),
     })
 }
 
@@ -230,7 +230,7 @@ fn parse_enum(item_enum: &ItemEnum) -> Option<RustDecl> {
         methods: Vec::new(),
         return_type: None,
         parameters: Vec::new(),
-        docstring: extract_docstring(&item_fn.attrs),
+        docstring: extract_docstring(&item_enum.attrs),
     })
 }
 
@@ -268,7 +268,7 @@ fn parse_trait(item_trait: &ItemTrait) -> Option<RustDecl> {
         methods,
         return_type: None,
         parameters: Vec::new(),
-        docstring: extract_docstring(&item_fn.attrs),
+        docstring: extract_docstring(&item_trait.attrs),
     })
 }