1 files changed, 5 insertions, 3 deletions
diff --git a/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt b/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt
index 2a45e986b0..d74e5278fa 100644
--- a/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt
+++ b/android/src/main/kotlin/net/mullvad/mullvadvpn/ConnectFragment.kt
@@ -43,9 +43,11 @@ class ConnectFragment : Fragment() {
         status = view.findViewById(R.id.connection_status)
 
         actionButton = ConnectActionButton(view)
-        actionButton.onConnect = { connect() }
-        actionButton.onCancel = { disconnect() }
-        actionButton.onDisconnect = { disconnect() }
+        actionButton.apply {
+            onConnect = { connect() }
+            onCancel = { disconnect() }
+            onDisconnect = { disconnect() }
+        }
 
         return view
     }