From 4b398d955e83aa89138a6e35ad4faebbcdb69a2f Mon Sep 17 00:00:00 2001
From: poorna2152 <poorna2152@gmail.com>
Date: Tue, 5 Nov 2024 14:27:44 +0530
Subject: [PATCH] Address review suggestion

---
 .../mi/plugin/VariableDeclarationAnalysisTask.java     | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/compiler-plugin/src/main/java/io/ballerina/stdlib/mi/plugin/VariableDeclarationAnalysisTask.java b/compiler-plugin/src/main/java/io/ballerina/stdlib/mi/plugin/VariableDeclarationAnalysisTask.java
index 11d368f..5b03709 100644
--- a/compiler-plugin/src/main/java/io/ballerina/stdlib/mi/plugin/VariableDeclarationAnalysisTask.java
+++ b/compiler-plugin/src/main/java/io/ballerina/stdlib/mi/plugin/VariableDeclarationAnalysisTask.java
@@ -96,11 +96,11 @@ private TypeSymbol getRawType(TypeSymbol typeDescriptor) {
     private List<ObjectTypeSymbol> getObjectTypeMembers(BallerinaUnionTypeSymbol unionTypeSymbol) {
         ArrayList<ObjectTypeSymbol> objectTypes = new ArrayList<>();
         for (TypeSymbol member : unionTypeSymbol.memberTypeDescriptors()) {
-            member = getRawType(member);
-            if (member.typeKind() == TypeDescKind.OBJECT) {
-                objectTypes.add((ObjectTypeSymbol) member);
-            } else if (member.typeKind() == TypeDescKind.UNION) {
-                objectTypes.addAll(getObjectTypeMembers((BallerinaUnionTypeSymbol) member));
+            TypeSymbol rawType = getRawType(member);
+            if (rawType.typeKind() == TypeDescKind.OBJECT) {
+                objectTypes.add((ObjectTypeSymbol) rawType);
+            } else if (rawType.typeKind() == TypeDescKind.UNION) {
+                objectTypes.addAll(getObjectTypeMembers((BallerinaUnionTypeSymbol) rawType));
             }
         }
         return objectTypes;