diff --git a/packages/core/src/components/modal/test/basic/index.html b/packages/core/src/components/modal/test/basic/index.html
index 0d3e5525b6..418164d945 100644
--- a/packages/core/src/components/modal/test/basic/index.html
+++ b/packages/core/src/components/modal/test/basic/index.html
@@ -46,7 +46,7 @@
       <ion-content>
         <h1>Content of doom</h1>
         <div>Here's some more content</div>
-        <ion-button>Dismiss Modal</ion-button>
+        <ion-button class="dismiss">Dismiss Modal</ion-button>
       </ion-content>
       `;
 
diff --git a/packages/core/src/components/modal/test/host-elements/index.html b/packages/core/src/components/modal/test/host-elements/index.html
index 973f09ed1d..ef0eea61d1 100644
--- a/packages/core/src/components/modal/test/host-elements/index.html
+++ b/packages/core/src/components/modal/test/host-elements/index.html
@@ -42,14 +42,14 @@
           <ion-content>
             <h1>Content of doom</h1>
             <div>Here's some more content</div>
-            <ion-button>Dismiss Modal</ion-button>
+            <ion-button class="dismiss">Dismiss Modal</ion-button>
           </ion-content>
         </ion-page>`;
 
           // listen for close event
         const button = this.querySelector('ion-button');
         button.addEventListener('click', async () => {
-          await this.querySelector('ion-modal-controller').dismiss();
+          await document.querySelector('ion-modal-controller').dismiss();
         });
       }
     }