diff --git a/source_gen/lib/src/library.dart b/source_gen/lib/src/library.dart
index 97c96e64..cd7dc601 100644
--- a/source_gen/lib/src/library.dart
+++ b/source_gen/lib/src/library.dart
@@ -41,8 +41,7 @@ class LibraryReader {
         ...element.libraryImports,
         // ignore: deprecated_member_use
         ...element.libraryExports,
-        // ignore: deprecated_member_use
-        ...element.parts,
+        ...element.definingCompilationUnit.parts,
       ];
 
   /// All of the declarations in this library annotated with [checker].