diff --git a/app/src/main/java/com/google/ai/edge/aicore/demo/kotlin/MainActivity.kt b/app/src/main/java/com/google/ai/edge/aicore/demo/kotlin/MainActivity.kt
index eb33184d..122a4c8f 100755
--- a/app/src/main/java/com/google/ai/edge/aicore/demo/kotlin/MainActivity.kt
+++ b/app/src/main/java/com/google/ai/edge/aicore/demo/kotlin/MainActivity.kt
@@ -141,6 +141,7 @@ class MainActivity : AppCompatActivity(), GenerationConfigDialog.OnConfigUpdateL
           contentAdapter.addContent(ContentAdapter.VIEW_TYPE_RESPONSE_ERROR, e.message!!)
           endGeneratingUi()
         }
+        inGenerating = false
       }
   }