1 files changed, 9 insertions, 4 deletions
diff --git a/iosApp/iosApp/Units/UnitsViewModel.swift b/iosApp/iosApp/Units/UnitsViewModel.swift
index 7ffcfd1..af6fe32 100644
--- a/iosApp/iosApp/Units/UnitsViewModel.swift
+++ b/iosApp/iosApp/Units/UnitsViewModel.swift
@@ -52,7 +52,8 @@ class UnitsViewModel: ObservableObject {
     }
     @Published var isEditing = false
     @Published var showDetails = false
-    @Published var unitsDisplayMode: UnitsDisplayMode = .list
+    @Published var showUserInfo = false
+    @Published var unitsDisplayMode: UnitsDisplayMode = .map
     @Published var units: [UnitInformation] = [] {
         didSet {
             markers = units.compactMap(Marker.companion.fromUnit)
@@ -127,8 +128,12 @@ class UnitsViewModel: ObservableObject {
     }
     
     func show(action: DeviceRow.Action, for unit: UnitInformation) {
-        detailsAction = action
-        detailsUnit = unit
-        showDetails = true
+        if action != .close {
+            detailsAction = action
+            detailsUnit = unit
+            showDetails = true
+        } else {
+            selectedUnit = nil
+        }
     }
 }