diff --git a/ShhShell/Host/HostsManager.swift b/ShhShell/Host/HostsManager.swift
index 8d53aaa..194954c 100644
--- a/ShhShell/Host/HostsManager.swift
+++ b/ShhShell/Host/HostsManager.swift
@@ -129,14 +129,13 @@ class HostsManager: ObservableObject, @unchecked Sendable {
 	}
 	
 	func updateTheme(_ theme: Theme) {
-		guard let index = themes.firstIndex(where: { $0.id == theme.id }) else {
+		if let index = themes.firstIndex(where: { $0.id == theme.id }) {
+			themes[index] = theme
+			if selectedTheme.id == theme.id {
+				selectedTheme = theme
+			}
+		} else {
 			themes.append(theme)
-			saveThemes()
-			return
-		}
-		themes[index] = theme
-		if selectedTheme.id == theme.id {
-			selectedTheme = theme
 		}
 		saveThemes()
 	}