Update Generate Key button text

1 files changed, 9 insertions, 8 deletions

diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/WireguardKeyFragment.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/WireguardKeyFragment.kt
index 91e368dbbb..bd74c2c8c5 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/WireguardKeyFragment.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ui/WireguardKeyFragment.kt
@@ -56,6 +56,7 @@ class WireguardKeyFragment : ServiceDependentFragment(OnNoService.GoToLaunchScre
                 field = value
                 updateKeyInformation()
                 updateStatusMessage()
+                updateGenerateKeyButtonText()
                 updateVerifyKeyButtonState()
             }
         }
@@ -241,6 +242,14 @@ class WireguardKeyFragment : ServiceDependentFragment(OnNoService.GoToLaunchScre
         generateKeyButton.setEnabled(actionState is ActionState.Idle && hasConnectivity)
     }
 
+    private fun updateGenerateKeyButtonText() {
+        if (keyStatus is KeygenEvent.NewKey) {
+            generateKeyButton.setText(R.string.wireguard_replace_key)
+        } else {
+            generateKeyButton.setText(R.string.wireguard_generate_key)
+        }
+    }
+
     private fun updateVerifyKeyButtonState() {
         val isIdle = actionState is ActionState.Idle
         val hasKey = keyStatus is KeygenEvent.NewKey
@@ -272,14 +281,6 @@ class WireguardKeyFragment : ServiceDependentFragment(OnNoService.GoToLaunchScre
         }
     }
 
-    private fun setGenerateButton() {
-        if (keyStatus is KeygenEvent.NewKey) {
-            generateKeyButton.setText(R.string.wireguard_replace_key)
-        } else {
-            generateKeyButton.setText(R.string.wireguard_generate_key)
-        }
-    }
-
     private suspend fun onGenerateKeyPress() {
         synchronized(this) {
             actionState = ActionState.Generating()