Modify logic in OSCompilerPluginUtil

ballerina-platform · Feb 19, 2025 · be812fd · be812fd
1 parent 0347400
commit be812fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/compiler-plugin/src/main/java/io/ballerina/stdlib/os/compiler/OSCompilerPluginUtil.java b/compiler-plugin/src/main/java/io/ballerina/stdlib/os/compiler/OSCompilerPluginUtil.java
@@ -47,6 +47,6 @@ private static boolean isOsModule(SyntaxNodeAnalysisContext context) {
     }
 
     public static Document getDocument(SyntaxNodeAnalysisContext context) {
-        return context.currentModule().document(context.node().location().lineRange().filePath());
+        return context.currentPackage().module(context.moduleId()).document(context.documentId());
     }
 }