1 files changed, 2 insertions, 0 deletions
diff --git a/llvm/lib/ExecutionEngine/Interpreter/UserInput.cpp b/llvm/lib/ExecutionEngine/Interpreter/UserInput.cpp
index 49fd5e4bc17..a1518d413e0 100644
--- a/llvm/lib/ExecutionEngine/Interpreter/UserInput.cpp
+++ b/llvm/lib/ExecutionEngine/Interpreter/UserInput.cpp
@@ -153,6 +153,7 @@ void Interpreter::loadModule(const string &Filename) {
          << ErrorMsg << "\n";
     return;
   }
+  CW.setModule(CurMod);  // Update Writer
 
   string RuntimeLib = getCurrentExecutablePath();
   if (!RuntimeLib.empty()) RuntimeLib += "/";
@@ -185,6 +186,7 @@ bool Interpreter::flushModule() {
     CurFrame = -1;
   }
 
+  CW.setModule(0);
   delete CurMod;
   CurMod = 0;
   ExitCode = 0;