diff --git a/src/ui.js b/src/ui.js
index 87e7cfc..332670f 100644
--- a/src/ui.js
+++ b/src/ui.js
@@ -2460,16 +2460,16 @@ export async function openUncloseaiEmbeddedModal() {
 							JSON.stringify(savedHistory),
 						);
 						// Update global chat history
-							chatHistory = [
-								{
-									role: "system",
-									content: getSystemMessage(),
-								},
-								...savedHistory,
-							];
-							restoreConversationHistory();
-						}
+						chatHistory = [
+							{
+								role: "system",
+								content: getSystemMessage(),
+							},
+							...savedHistory,
+						];
+						restoreConversationHistory();
 					}
+					messageDiv.remove();
 				};
 			})(index);