Update notification based on real tunnel state

1 files changed, 3 insertions, 3 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/ForegroundNotificationManager.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/ForegroundNotificationManager.kt
index f42a48ce8d..dfb153bbc2 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/ForegroundNotificationManager.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ForegroundNotificationManager.kt
@@ -26,8 +26,8 @@ class ForegroundNotificationManager(val service: Service, val connectionProxy: C
     private val notificationManager =
         service.getSystemService(Context.NOTIFICATION_SERVICE) as NotificationManager
 
-    private val listenerId = connectionProxy.onUiStateChange.subscribe { uiState ->
-        tunnelState = uiState
+    private val listenerId = connectionProxy.onStateChange.subscribe { state ->
+        tunnelState = state
     }
 
     private var reconnecting = false
@@ -147,7 +147,7 @@ class ForegroundNotificationManager(val service: Service, val connectionProxy: C
 
     fun onDestroy() {
         listenerId?.let { listener ->
-            connectionProxy.onUiStateChange.unsubscribe(listener)
+            connectionProxy.onStateChange.unsubscribe(listener)
         }
 
         service.apply {