From b46c337bcf7eb71d4990d886af94b19e01afe345 Mon Sep 17 00:00:00 2001
From: anutosh491 <andersonbhat491@gmail.com>
Date: Tue, 20 Feb 2024 16:58:29 +0530
Subject: [PATCH] Freeing variables correctly through visit_Return

---
 src/libasr/pass/replace_symbolic.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/libasr/pass/replace_symbolic.cpp b/src/libasr/pass/replace_symbolic.cpp
index 96a1469765..cefb081630 100644
--- a/src/libasr/pass/replace_symbolic.cpp
+++ b/src/libasr/pass/replace_symbolic.cpp
@@ -985,7 +985,9 @@ class ReplaceSymbolicVisitor : public PassUtils::PassVisitor<ReplaceSymbolicVisi
                     ASRUtils::EXPR(ASR::make_Var_t(al, x.base.base.loc, symbol))));
             }
             symbolic_vars_to_free.clear();
-            pass_result.push_back(al, ASRUtils::STMT(ASR::make_Return_t(al, x.base.base.loc)));
+            for (ASR::symbol_t* symbol : symbolic_vars_to_omit) {
+                symbolic_vars_to_free.insert(symbol);
+            }
         }
     }
 };