1 files changed, 3 insertions, 2 deletions
diff --git a/tests/robolectric/src/com/android/systemui/plugin/globalactions/wallet/WalletPanelViewControllerTest.java b/tests/robolectric/src/com/android/systemui/plugin/globalactions/wallet/WalletPanelViewControllerTest.java
index 89b25f6..34c83eb 100644
--- a/tests/robolectric/src/com/android/systemui/plugin/globalactions/wallet/WalletPanelViewControllerTest.java
+++ b/tests/robolectric/src/com/android/systemui/plugin/globalactions/wallet/WalletPanelViewControllerTest.java
@@ -240,13 +240,14 @@ public class WalletPanelViewControllerTest {
         mViewController.queryWalletCards();
 
         verify(mWalletClient, never()).getWalletCards(any(), any(), any());
-        assertThat(errorView.getVisibility()).isEqualTo(VISIBLE);
+        assertThat(walletView.getVisibility()).isEqualTo(GONE);
 
         mViewController.onDeviceLockStateChanged(false);
         mViewController.onDeviceLockStateChanged(true);
 
         verify(mWalletClient).getWalletCards(any(), any(), any());
-        assertThat(errorView.getVisibility()).isEqualTo(View.GONE);
+        assertThat(walletView.getVisibility()).isEqualTo(VISIBLE);
+        assertThat(errorView.getVisibility()).isEqualTo(GONE);
     }
 
     @Test