diff --git a/VoiceInk/Services/AIEnhancementService.swift b/VoiceInk/Services/AIEnhancementService.swift
index f008e2b..25f1853 100644
--- a/VoiceInk/Services/AIEnhancementService.swift
+++ b/VoiceInk/Services/AIEnhancementService.swift
@@ -182,17 +182,8 @@ class AIEnhancementService: ObservableObject {
                 let result = try await aiService.enhanceWithOllama(text: formattedText, systemPrompt: systemMessage)
                 let filteredResult = AIEnhancementOutputFilter.filter(result)
                 return filteredResult
-            } catch let error as LocalAIError {
-                switch error {
-                case .serviceUnavailable:
-                    throw EnhancementError.notConfigured
-                case .modelNotFound:
-                    throw EnhancementError.enhancementFailed
-                case .serverError:
-                    throw EnhancementError.serverError
-                default:
-                    throw EnhancementError.enhancementFailed
-                }
+            } catch {
+                throw EnhancementError.customError(error.localizedDescription)
             }
         }