diff --git a/e2e/login.ts b/e2e/login.ts
index 5f2b3d91..097a08f4 100644
--- a/e2e/login.ts
+++ b/e2e/login.ts
@@ -13,6 +13,6 @@ export function login(route: string) {
     await expect(page).toHaveURL(
       "http://localhost:8080/index.php/apps/memories" + route
     );
-    await page.waitForSelector('img[src*="api/image/preview"]');
+    await page.waitForSelector("img.ximg");
   };
 }