Index: pkg/analysis_server/lib/src/services/correction/assist_internal.dart |
diff --git a/pkg/analysis_server/lib/src/services/correction/assist_internal.dart b/pkg/analysis_server/lib/src/services/correction/assist_internal.dart |
index 73a4beef69a9690582eeb22eeb0a5d0ea316545b..0e3cfe935d444f7c1fdea67f3bce66a2c73b50fe 100644 |
--- a/pkg/analysis_server/lib/src/services/correction/assist_internal.dart |
+++ b/pkg/analysis_server/lib/src/services/correction/assist_internal.dart |
@@ -18,6 +18,7 @@ import 'package:analysis_server/src/services/correction/statement_analyzer.dart' |
import 'package:analysis_server/src/services/correction/util.dart'; |
import 'package:analysis_server/src/services/search/hierarchy.dart'; |
import 'package:analyzer/dart/ast/ast.dart'; |
+import 'package:analyzer/dart/ast/standard_resolution_map.dart'; |
import 'package:analyzer/dart/ast/token.dart'; |
import 'package:analyzer/dart/ast/visitor.dart'; |
import 'package:analyzer/dart/element/element.dart'; |
@@ -72,7 +73,8 @@ class AssistProcessor { |
unit = dartContext.unit; |
unitElement = dartContext.unit.element; |
// library |
- unitLibraryElement = dartContext.unit.element.library; |
+ unitLibraryElement = |
+ resolutionMap.elementForCompilationUnit(dartContext.unit).library; |
unitLibraryFile = unitLibraryElement.source.fullName; |
unitLibraryFolder = dirname(unitLibraryFile); |
// selection |