diff --git a/app/page.tsx b/app/page.tsx
index 63df21c..b4c463e 100644
--- a/app/page.tsx
+++ b/app/page.tsx
@@ -150,6 +150,13 @@ export default function Home() {
     }
   }, [hasApiKey, apiKeyStatus, justDidReset])
 
+  // Ensure model load status is set correctly when model is selected
+  useEffect(() => {
+    if (hasModel && modelLoadStatus !== 'success') {
+      setModelLoadStatus('success')
+    }
+  }, [hasModel, modelLoadStatus])
+
   const handleApiKeyValidated = (isValid: boolean, key?: string) => {
     if (isValid && key) {
       setApiKeyStatus('success')