diff --git a/llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java b/llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
index e80fbc8f..4f01f2be 100644
--- a/llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
+++ b/llm/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo/MainActivity.java
@@ -845,6 +845,7 @@ public void run() {
                           onModelRunStopped();
                         }
                       });
+                  mModule.resetContext();
                   ETLogging.getInstance().log("Inference completed");
                 }
               };