diff --git a/src/custom_widgets/message_widget.py b/src/custom_widgets/message_widget.py
index e7a5233..606a175 100644
--- a/src/custom_widgets/message_widget.py
+++ b/src/custom_widgets/message_widget.py
@@ -105,7 +105,7 @@ class code_block(Gtk.Box):
         title_box = Gtk.Box(margin_start=12, margin_top=3, margin_bottom=3, margin_end=3)
         title_box.append(Gtk.Label(label=self.language.get_name() if self.language else _("Code Block"), hexpand=True, xalign=0))
         copy_button = Gtk.Button(icon_name="edit-copy-symbolic", css_classes=["flat", "circular"], tooltip_text=_("Copy Message"))
-        copy_button.connect("clicked", lambda *_: self.on_copy)
+        copy_button.connect("clicked", lambda *_: self.on_copy())
         title_box.append(copy_button)
         self.append(title_box)
         self.append(Gtk.Separator())
diff --git a/src/window.py b/src/window.py
index 6ab0b3e..4424e31 100644
--- a/src/window.py
+++ b/src/window.py
@@ -525,6 +525,7 @@ Generate a title following these rules:
 
     def run_message(self, data:dict, message_element:message_widget.message, chat:chat_widget.chat):
         logger.debug("Running message")
+        self.save_history(chat)
         chat.busy = True
         self.chat_list_box.get_tab_by_name(chat.get_name()).spinner.set_visible(True)
         if len(data['messages']) == 1 and chat.get_name().startswith(_("New Chat")):