diff --git a/src/lib/Storage.svelte b/src/lib/Storage.svelte
index 4386a34..b7db875 100644
--- a/src/lib/Storage.svelte
+++ b/src/lib/Storage.svelte
@@ -351,7 +351,7 @@
   }
 
   export const updateChatImages = async (chatId: number, chat: Chat) => {
-    const messages = getMessages(chatId)
+    const messages = chat.messages
     for (let i = 0; i < messages.length; i++) {
       const m = messages[i]
       if (m.image) m.image = await setImage(chatId, m.image)