diff --git a/e2e-tests/cypress/tests/support/ui/file_preview.js b/e2e-tests/cypress/tests/support/ui/file_preview.js
index 67adfd0154e..e4d49dadfba 100644
--- a/e2e-tests/cypress/tests/support/ui/file_preview.js
+++ b/e2e-tests/cypress/tests/support/ui/file_preview.js
@@ -11,7 +11,7 @@ Cypress.Commands.add('uiGetFileUploadPreview', () => {
 
 Cypress.Commands.add('uiWaitForFileUploadPreview', () => {
     cy.waitUntil(() => cy.uiGetFileUploadPreview().then((el) => {
-        return el.find('.post-image.normal').length > 0;
+        return el.find('.post-image__thumbnail').length > 0;
     }));
 });