diff --git a/src/Webform.unit.js b/src/Webform.unit.js
index 31b0a34b5d..82698fa6fd 100644
--- a/src/Webform.unit.js
+++ b/src/Webform.unit.js
@@ -1363,7 +1363,7 @@ describe('Webform tests', function() {
                   setTimeout(() => {
                     const alertErrorMessagesAfterInputtingInvalidValues = document
                       .querySelector(`.editgrid-row-modal-${editGrid.id}`)
-                      .querySelectorAll('[ref="messageRef"]');
+                      .querySelectorAll('[ref="messageContainer"]');
 
                     assert.equal(alertErrorMessagesAfterInputtingInvalidValues.length, 2);
                     document.body.innerHTML = '';