From 57d484f9b8e7f6a504966a901174358a42fa932a Mon Sep 17 00:00:00 2001
From: Googler <copybara-servicebot@google.com>
Date: Thu, 10 Oct 2024 09:43:47 -0700
Subject: [PATCH] Migration for analyzer APIs.

PiperOrigin-RevId: 684477097
---
 lib/src/builder.dart | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/src/builder.dart b/lib/src/builder.dart
index db956fbe..309e5b0a 100644
--- a/lib/src/builder.dart
+++ b/lib/src/builder.dart
@@ -474,8 +474,8 @@ class _MockTargetGatherer {
     final mockTargets = <_MockTarget>{};
 
     final possiblyAnnotatedElements = [
-      ...entryLib.libraryExports,
-      ...entryLib.libraryImports,
+      ...entryLib.definingCompilationUnit.libraryExports,
+      ...entryLib.definingCompilationUnit.libraryImports,
       ...entryLib.topLevelElements,
     ];