diff --git a/packages/primevue/src/dialog/Dialog.spec.js b/packages/primevue/src/dialog/Dialog.spec.js
index 4e87281f4..2f4a5c8b0 100644
--- a/packages/primevue/src/dialog/Dialog.spec.js
+++ b/packages/primevue/src/dialog/Dialog.spec.js
@@ -79,7 +79,7 @@ describe('closable', () => {
 
         await wrapper.setProps({ visible: true });
 
-        const icon = wrapper.find('.p-dialog-header-close-icon');
+        const icon = wrapper.find('.p-dialog-close-button .p-button-icon');
 
         expect(icon.classes()).toContain('pi-discord');
     });
@@ -111,7 +111,7 @@ describe('maximizable', () => {
         await wrapper.setProps({ visible: true });
         await wrapper.setData({ maximized: false });
 
-        const icon = wrapper.find('.p-dialog-header-maximize-icon');
+        const icon = wrapper.find('.p-dialog-maximize-button .p-button-icon');
 
         expect(icon.classes()).toContain('pi-discord');