From 7cf3fcb4ef2983f7046ca08a68e96ea74cefcdaa Mon Sep 17 00:00:00 2001
From: Dan Bucholtz <dan.bucholtz@gmail.com>
Date: Tue, 9 Jan 2018 11:19:25 -0600
Subject: [PATCH] test(modal): fix tests

---
 packages/core/src/components/modal/test/basic/index.html      | 2 +-
 .../core/src/components/modal/test/host-elements/index.html   | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

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();
         });
       }
     }