diff --git a/VoiceInk/Services/AIEnhancementService.swift b/VoiceInk/Services/AIEnhancementService.swift
index adfd1e6..c807bf9 100644
--- a/VoiceInk/Services/AIEnhancementService.swift
+++ b/VoiceInk/Services/AIEnhancementService.swift
@@ -455,7 +455,7 @@ class AIEnhancementService: ObservableObject {
         guard retryCount < maxRetries - 1 else { return false }
         
         switch error {
-        case .rateLimitExceeded, .serverError, .networkError:
+        case .rateLimitExceeded, .serverError:
             return true
         default:
             return false
diff --git a/VoiceInk/Services/AIService.swift b/VoiceInk/Services/AIService.swift
index 5d22126..0ca586a 100644
--- a/VoiceInk/Services/AIService.swift
+++ b/VoiceInk/Services/AIService.swift
@@ -535,5 +535,3 @@ class AIService: ObservableObject {
 extension Notification.Name {
     static let aiProviderKeyChanged = Notification.Name("aiProviderKeyChanged")
 } 
-
-