diff --git a/static/main-v0.4.wasm b/static/main-v0.4.wasm
new file mode 100644
index 0000000..a580aa9
Binary files /dev/null and b/static/main-v0.4.wasm differ
diff --git a/views/layouts/main.html b/views/layouts/main.html
index 05fbc9e..0e1869b 100644
--- a/views/layouts/main.html
+++ b/views/layouts/main.html
@@ -27,7 +27,7 @@
     <script src="wasm_exec.js"></script>
     <script>
         const go = new Go();
-        WebAssembly.instantiateStreaming(fetch("main-v0.3.wasm"), go.importObject).then((result) => {
+        WebAssembly.instantiateStreaming(fetch("main-v0.4.wasm"), go.importObject).then((result) => {
             go.run(result.instance);
         });
     </script>
diff --git a/views/partials/chat-input.html b/views/partials/chat-input.html
index e415bec..07f62be 100644
--- a/views/partials/chat-input.html
+++ b/views/partials/chat-input.html
@@ -79,11 +79,10 @@
     function toggleSendButton() {
         // check if generate-multiple-messages exists 
         var generateMultipleMessages = document.getElementById('generate-multiple-messages');
-        var disabled = textarea.value.trim().length === 0 || document.getElementsByClassName('selected icon-llm').length === 0 || generateMultipleMessages !== null;
-        document.getElementById('chat-input-send-btn').disabled = disabled;
+        document.getElementById('chat-input-send-btn').disabled = textarea.value.trim().length === 0 || document.getElementsByClassName('selected icon-llm').length === 0 || generateMultipleMessages !== null;
         // Do the same for all element with an id that start with redo-button- or edit-button-
         const buttons = document.querySelectorAll('[id^="redo-button-"], [id^="edit-button-"]');
-        buttons.forEach(button => button.disabled = disabled);
+        buttons.forEach(button => button.disabled = document.getElementsByClassName('selected icon-llm').length === 0 || generateMultipleMessages !== null);
     }
 
     function onClickSendButton() {