diff --git a/firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/MainActivity.kt b/firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/MainActivity.kt
index d9f428f91fb..8165362dacd 100644
--- a/firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/MainActivity.kt
+++ b/firebase-appdistribution/test-app/src/main/java/com/googletest/firebase/appdistribution/testapp/MainActivity.kt
@@ -122,13 +122,15 @@ class MainActivity : AppCompatActivity() {
             executorService.execute {
                 firebaseAppDistribution
                     .checkForNewRelease()
-                    .addOnSuccessListener { release = it }
+                    .addOnSuccessListener {
+                        release = it
+                        setupUI(
+                            isSignedIn = firebaseAppDistribution.isTesterSignedIn,
+                            isUpdateAvailable = release != null,
+                            release = release)
+                    }
                     .addOnFailureListener { failureListener(it) }
             }
-            setupUI(
-                isSignedIn = firebaseAppDistribution.isTesterSignedIn,
-                isUpdateAvailable = release != null,
-                release = release)
         }
 
         signOutButton.setOnClickListener {