From 0c4f9465dffd8bf21ef066665247b50ad1f75727 Mon Sep 17 00:00:00 2001
From: ALEXks <alexks179@gmail.com>
Date: Thu, 19 Feb 2026 08:34:58 +0300
Subject: [PATCH] fixed getModuleSymbols

---
 src/Utils/module_utils.cpp | 4 ++--
 src/Utils/module_utils.h   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Utils/module_utils.cpp b/src/Utils/module_utils.cpp
index df3479f..74c71e2 100644
--- a/src/Utils/module_utils.cpp
+++ b/src/Utils/module_utils.cpp
@@ -325,7 +325,7 @@ static void getModuleSymbols(SgStatement* func, set<SgSymbol*>& symbs)
     }
 }
 
-set<SgSymbol*> getModuleSymbols(SgStatement *func)
+const set<SgSymbol*>& getModuleSymbols(SgStatement *func)
 {
     auto it = symbolsForFunc.find(func->symbol()->identifier());
     if (it != symbolsForFunc.end())
@@ -345,7 +345,7 @@ set<SgSymbol*> getModuleSymbols(SgStatement *func)
     }
 
     symbolsForFunc[func->symbol()->identifier()] = symbs;
-    return symbs;
+    return symbolsForFunc[func->symbol()->identifier()];
 }
 
 static void findSymbol(SgStatement* func, const string& varName, const string& locName,
diff --git a/src/Utils/module_utils.h b/src/Utils/module_utils.h
index dba36e9..e78b8e5 100644
--- a/src/Utils/module_utils.h
+++ b/src/Utils/module_utils.h
@@ -1,7 +1,7 @@
 #pragma once
 
 std::string correctSymbolModuleName(const std::string& origFull);
-std::set<SgSymbol*> getModuleSymbols(SgStatement* func);
+const std::set<SgSymbol*>& getModuleSymbols(SgStatement* func);
 void getModulesAndFunctions(SgFile* file, std::vector<SgStatement*>& modulesAndFunctions);
 void findModulesInFile(SgFile* file, std::vector<SgStatement*>& modules);
 std::map<std::string, std::set<std::string>> createMapOfModuleUses(SgFile* file);