diff --git a/components/ApiKeySetup.tsx b/components/ApiKeySetup.tsx
index 40c2497..e74fe32 100644
--- a/components/ApiKeySetup.tsx
+++ b/components/ApiKeySetup.tsx
@@ -122,9 +122,11 @@ export const ApiKeySetup: React.FC<ApiKeySetupProps> = ({ onKeySave }) => {
     try {
       const result = await testApiKey(provider, localApiKey, endpoint);
       
-      if (result.success && result.models && result.models.length > 0) {
-        setAvailableModels(result.models);
-        setModel(result.models[0].id);
+      if (result.success) {
+        if (result.models && result.models.length > 0) {
+          setAvailableModels(result.models);
+          setModel(result.models[0].id);
+        }
         setTestStatus('success');
         setError(null);
       } else {