diff --git a/ui/index.html b/ui/index.html
index d2b22951..8debceaa 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -877,6 +877,8 @@ async function makeImage() {
 
     makeImageBtn.innerHTML = 'Make Image'
     makeImageBtn.disabled = false
+    makeImageBtn.style.display = 'block'
+    stopImageBtn.style.display = 'none'
 
     if (isSoundEnabled()) {
         playSound()