diff --git a/app/src/test/java/protect/card_locker/LoyaltyCardViewActivityTest.java b/app/src/test/java/protect/card_locker/LoyaltyCardViewActivityTest.java
index e9fd035fe0..bf5548f09f 100644
--- a/app/src/test/java/protect/card_locker/LoyaltyCardViewActivityTest.java
+++ b/app/src/test/java/protect/card_locker/LoyaltyCardViewActivityTest.java
@@ -2,6 +2,7 @@
 
 import android.app.Activity;
 import android.app.DatePickerDialog;
+import android.app.Dialog;
 import android.content.Context;
 import android.content.Intent;
 import android.content.SharedPreferences;
@@ -725,9 +726,9 @@ public void startWithLoyaltyCardNoExpirySetExpiry() throws IOException {
 
         shadowOf(getMainLooper()).idle();
 
-        DatePickerDialog datePickerDialog = (DatePickerDialog) (ShadowDialog.getLatestDialog());
+        Dialog datePickerDialog = ShadowDialog.getLatestDialog();
         assertNotNull(datePickerDialog);
-        datePickerDialog.getButton(DatePickerDialog.BUTTON_POSITIVE).performClick();
+        datePickerDialog.findViewById(com.google.android.material.R.id.confirm_button).performClick();
 
         shadowOf(getMainLooper()).idle();