diff --git a/desktop/tests/e2e/needs-restart-screenshots.spec.ts b/desktop/tests/e2e/needs-restart-screenshots.spec.ts
index 508a5d91e..547ca22cb 100644
--- a/desktop/tests/e2e/needs-restart-screenshots.spec.ts
+++ b/desktop/tests/e2e/needs-restart-screenshots.spec.ts
@@ -261,7 +261,7 @@ test.describe("restart-diff screenshots", () => {
     await gotoAgentsView(page);
 
     const personaCard = page.getByTestId(
-      `persona-agent-row-${PERSONA_AGENT.personaId}`,
+      `managed-agent-${PERSONA_AGENT.pubkey}`,
     );
     await expect(personaCard).toBeVisible({ timeout: 10_000 });
     await expect(