diff --git a/docs/src/demos/Extensions/Blockquote/index.spec.js b/docs/src/demos/Extensions/Blockquote/index.spec.js
index 0e77f053..af396cb1 100644
--- a/docs/src/demos/Extensions/Blockquote/index.spec.js
+++ b/docs/src/demos/Extensions/Blockquote/index.spec.js
@@ -20,6 +20,8 @@ context('/api/extensions/blockquote', () => {
       cy.get('.ProseMirror blockquote').should('not.exist')
       cy.get('.demo__preview button:first').click({ force: true })
       cy.get('.ProseMirror').contains('blockquote', 'Example Text')
+
+      cy.get('.ProseMirror').type('{selectall}', {force: true})
       cy.get('.demo__preview button:first').click({ force: true })
       cy.get('.ProseMirror blockquote').should('not.exist')
     })
@@ -33,6 +35,8 @@ context('/api/extensions/blockquote', () => {
       cy.get('.ProseMirror blockquote').should('not.exist')
       cy.get('.ProseMirror').type('{meta}{shift}9', {force: true})
       cy.get('.ProseMirror').contains('blockquote', 'Example Text')
+
+      cy.get('.ProseMirror').type('{selectall}', {force: true})
       cy.get('.ProseMirror').type('{meta}{shift}9', {force: true})
       cy.get('.ProseMirror blockquote').should('not.exist')
     })
diff --git a/docs/src/demos/Extensions/CodeBlock/index.spec.js b/docs/src/demos/Extensions/CodeBlock/index.spec.js
index c9cb0245..51c40a31 100644
--- a/docs/src/demos/Extensions/CodeBlock/index.spec.js
+++ b/docs/src/demos/Extensions/CodeBlock/index.spec.js
@@ -18,6 +18,8 @@ context('/api/extensions/code-block', () => {
     it('the button should toggle the code block', () => {
       cy.get('.demo__preview button:first').click({ force: true })
       cy.get('.ProseMirror').contains('pre', 'Example Text')
+
+      cy.get('.ProseMirror').type('{selectall}', {force: true})
       cy.get('.demo__preview button:first').click({ force: true })
       cy.get('.ProseMirror pre').should('not.exist')
     })
@@ -30,6 +32,8 @@ context('/api/extensions/code-block', () => {
     it('the keyboard shortcut should toggle the code block', () => {
       cy.get('.ProseMirror').type('{control}{shift}\\', {force: true})
       cy.get('.ProseMirror').contains('pre', 'Example Text')
+
+      cy.get('.ProseMirror').type('{selectall}', {force: true})
       cy.get('.ProseMirror').type('{control}{shift}\\', {force: true})
       cy.get('.ProseMirror pre').should('not.exist')
     })