Small android style fix

mullvad · Dec 14, 2023 · e01260b · e01260b
1 parent 398e8c4
commit e01260b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/ConnectViewModel.kt b/android/app/src/main/kotlin/net/mullvad/mullvadvpn/viewmodel/ConnectViewModel.kt
@@ -83,7 +83,7 @@ class ConnectViewModel(
                     notifications,
                     tunnelUiState,
                     tunnelRealState,
-                    lastKnownLocation,
+                    lastKnownDisconnectedLocation,
                     accountExpiry,
                     deviceName ->
                     if (tunnelRealState.isTunnelErrorStateDueToExpiredAccount()) {
@@ -93,11 +93,11 @@ class ConnectViewModel(
                         location =
                             when (tunnelRealState) {
                                 is TunnelState.Disconnected -> tunnelRealState.location()
-                                        ?: lastKnownLocation
+                                        ?: lastKnownDisconnectedLocation
                                 is TunnelState.Connecting -> tunnelRealState.location
                                 is TunnelState.Connected -> tunnelRealState.location
-                                is TunnelState.Disconnecting -> lastKnownLocation
-                                is TunnelState.Error -> lastKnownLocation
+                                is TunnelState.Disconnecting -> lastKnownDisconnectedLocation
+                                is TunnelState.Error -> lastKnownDisconnectedLocation
                             },
                         relayLocation = relayLocation,
                         tunnelUiState = tunnelUiState,

diff --git a/android/lib/model/src/main/kotlin/net/mullvad/mullvadvpn/model/TunnelState.kt b/android/lib/model/src/main/kotlin/net/mullvad/mullvadvpn/model/TunnelState.kt
@@ -26,10 +26,10 @@ sealed class TunnelState : Parcelable {
 
     fun location(): GeoIpLocation? {
         return when (this) {
-            is Connected -> this.location
-            is Connecting -> this.location
+            is Connected -> location
+            is Connecting -> location
             is Disconnecting -> null
-            is Disconnected -> this.location
+            is Disconnected -> location
             is Error -> null
         }
     }