diff --git a/tests/javascripts/preventDuplicateFormSubmissions.test.js b/tests/javascripts/preventDuplicateFormSubmissions.test.js
index 181b78853..f6e1dd462 100644
--- a/tests/javascripts/preventDuplicateFormSubmissions.test.js
+++ b/tests/javascripts/preventDuplicateFormSubmissions.test.js
@@ -19,7 +19,7 @@ describe('Prevent duplicate form submissions', () => {
     // set up DOM
     document.body.innerHTML = `
       <form action="/" method="post">
-        <button class="button" type="submit">Continue</button>
+        <button class="govuk-button" type="submit">Continue</button>
       </form>`;
 
     form = document.querySelector('form');
diff --git a/tests/javascripts/previewPane.test.js b/tests/javascripts/previewPane.test.js
index dd25031dc..c4ed22425 100644
--- a/tests/javascripts/previewPane.test.js
+++ b/tests/javascripts/previewPane.test.js
@@ -73,11 +73,11 @@ describe('Preview pane', () => {
                   <input autocomplete="off" class="form-control form-control-1-1 " id="search" name="search" required="" rows="8" type="search" value="">
                 </div>
               </div>
-            </div> 
+            </div>
           </div>
         </div>
         <div class="page-footer">
-          <button type="submit" class="button">Preview</button>
+          <button type="submit" class="govuk-button">Preview</button>
         </div>
       </form>`;
 
diff --git a/tests/javascripts/stick-to-window-when-scrolling.test.js b/tests/javascripts/stick-to-window-when-scrolling.test.js
index 32bacaf9f..353adc45d 100644
--- a/tests/javascripts/stick-to-window-when-scrolling.test.js
+++ b/tests/javascripts/stick-to-window-when-scrolling.test.js
@@ -91,7 +91,7 @@ describe("Stick to top/bottom of window when scrolling", () => {
                 </div>
               </div>
               <div class="page-footer">
-                <button type="submit" class="button">Continue</button>
+                <button type="submit" class="govuk-button">Continue</button>
               </div>
             </form>
           </main>
@@ -701,7 +701,7 @@ describe("Stick to top/bottom of window when scrolling", () => {
             </div>
             <div class="page-footer js-stick-at-bottom-when-scrolling">
               <form method="post" action="">
-                  <button type="submit" class="button">Send 1 email </button>
+                  <button type="submit" class="govuk-button">Send 1 email </button>
               </form>
             </div>
           </div>