diff --git a/cypress/support/commands.js b/cypress/support/commands.js
index 4927ac8..1d8f960 100644
--- a/cypress/support/commands.js
+++ b/cypress/support/commands.js
@@ -16,16 +16,16 @@ Cypress.Commands.add(
 
 Cypress.Commands.add('previewOn' , () => {
     cy.get('.EasyMDEContainer .editor-preview').should('not.be.visible');
-    cy.get('.EasyMDEContainer .editor-toolbar button.preview').should('not.have.class', 'active');
-    cy.get('.EasyMDEContainer .editor-toolbar button.preview').click();
-    cy.get('.EasyMDEContainer .editor-toolbar button.preview').should('have.class', 'active');
+    cy.get('.EasyMDEContainer .editor-toolbar button.mde-preview').should('not.have.class', 'active');
+    cy.get('.EasyMDEContainer .editor-toolbar button.mde-preview').click();
+    cy.get('.EasyMDEContainer .editor-toolbar button.mde-preview').should('have.class', 'active');
     cy.get('.EasyMDEContainer .editor-preview').should('be.visible');
 });
 
 Cypress.Commands.add('previewOff' , () => {
     cy.get('.EasyMDEContainer .editor-preview').should('be.visible');
-    cy.get('.EasyMDEContainer .editor-toolbar button.preview').should('have.class', 'active');
-    cy.get('.EasyMDEContainer .editor-toolbar button.preview').click();
-    cy.get('.EasyMDEContainer .editor-toolbar button.preview').should('not.have.class', 'active');
+    cy.get('.EasyMDEContainer .editor-toolbar button.mde-preview').should('have.class', 'active');
+    cy.get('.EasyMDEContainer .editor-toolbar button.mde-preview').click();
+    cy.get('.EasyMDEContainer .editor-toolbar button.mde-preview').should('not.have.class', 'active');
     cy.get('.EasyMDEContainer .editor-preview').should('not.be.visible');
 });