diff --git a/Demo/App/APIProvidedView.swift b/Demo/App/APIProvidedView.swift
index 7dc61b5d..9771e1fb 100644
--- a/Demo/App/APIProvidedView.swift
+++ b/Demo/App/APIProvidedView.swift
@@ -51,6 +51,7 @@ struct APIProvidedView: View {
         .onChange(of: apiKey) { newApiKey in
             let client = OpenAI(apiToken: newApiKey)
             chatStore.openAIClient = client
+            imageStore.openAIClient = client
             miscStore.openAIClient = client
         }
     }