2 files changed, 3 insertions, 3 deletions
diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/dataproxy/ConnectionProxy.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/dataproxy/ConnectionProxy.kt
index 8922344de5..8782493b4c 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/dataproxy/ConnectionProxy.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/dataproxy/ConnectionProxy.kt
@@ -36,7 +36,7 @@ class ConnectionProxy(val parentActivity: MainActivity) {
     var onUiStateChange: ((TunnelState) -> Unit)? = null
 
     fun connect() {
-        uiState = TunnelState.Connecting()
+        uiState = TunnelState.Connecting(null)
 
         cancelActiveAction()
 
diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/model/TunnelState.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/model/TunnelState.kt
index 006de053c8..2551cfe978 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/model/TunnelState.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/model/TunnelState.kt
@@ -2,8 +2,8 @@ package net.mullvad.mullvadvpn.model
 
 sealed class TunnelState() {
     class Disconnected() : TunnelState()
-    class Connecting() : TunnelState()
-    class Connected() : TunnelState()
+    class Connecting(val location: GeoIpLocation?) : TunnelState()
+    class Connected(val location: GeoIpLocation) : TunnelState()
     class Disconnecting() : TunnelState()
     class Blocked() : TunnelState()
 }