diff --git a/playwright/e2e/settings/encryption-user-tab/advanced.spec.ts b/playwright/e2e/settings/encryption-user-tab/advanced.spec.ts
index 17ff481aa6..2b459c2e11 100644
--- a/playwright/e2e/settings/encryption-user-tab/advanced.spec.ts
+++ b/playwright/e2e/settings/encryption-user-tab/advanced.spec.ts
@@ -63,7 +63,9 @@ test.describe("Advanced section in Encryption tab", () => {
             await expect(section.getByRole("button", { name: "Reset cryptographic identity" })).toBeVisible();
 
             // After resetting the identity, the user should set up a new recovery key
-            await expect(page.getByRole("checkbox", { name: "Allow key storage" })).not.toBeChecked();
+            await expect(
+                util.getEncryptionRecoverySection().getByRole("button", { name: "Set up recovery" }),
+            ).toBeVisible();
 
             await checkDeviceIsCrossSigned(app);
         },