From 68ee52b77daf9b6d7510995032b1ce2cd1dd754b Mon Sep 17 00:00:00 2001
From: Lana243 <lana0771711@gmail.com>
Date: Mon, 24 Apr 2023 16:40:02 +0300
Subject: [PATCH] Fix createLibcWrapper

---
 lib/Runner/run_klee.cpp | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/lib/Runner/run_klee.cpp b/lib/Runner/run_klee.cpp
index 74f04438dc..ed357a9ccb 100644
--- a/lib/Runner/run_klee.cpp
+++ b/lib/Runner/run_klee.cpp
@@ -1087,10 +1087,11 @@ createLibCWrapper(std::vector<std::unique_ptr<llvm::Module>> &userModules,
   auto &ctx = userModules[0]->getContext();
   Function *userMainFn = nullptr;
   for (auto &module : userModules) {
-    userMainFn = module->getFunction(intendedFunction);
-    if (userMainFn) {
+    Function *func = module->getFunction(intendedFunction);
+    if (func) {
       // Rename entry point using a prefix
-      userMainFn->setName("__user_" + intendedFunction);
+      func->setName("__user_" + intendedFunction);
+      userMainFn = func;
     }
   }