diff --git a/docs/src/demos/Extensions/Heading/index.spec.js b/docs/src/demos/Extensions/Heading/index.spec.js
new file mode 100644
index 00000000..065b0c79
--- /dev/null
+++ b/docs/src/demos/Extensions/Heading/index.spec.js
@@ -0,0 +1,5 @@
+context('/api/extensions/heading', () => {
+  beforeEach(() => {
+    cy.visit('/api/extensions/heading')
+  })
+})
\ No newline at end of file
diff --git a/docs/src/demos/Extensions/Heading/index.vue b/docs/src/demos/Extensions/Heading/index.vue
new file mode 100644
index 00000000..3be306eb
--- /dev/null
+++ b/docs/src/demos/Extensions/Heading/index.vue
@@ -0,0 +1,61 @@
+<template>
+  <div v-if="editor">
+    <button @click="editor.focus().heading({ level: 1 })" :class="{ 'is-active': editor.isActive('heading', { level: 1 }) }">
+      h1
+    </button>
+    <button @click="editor.focus().heading({ level: 2 })" :class="{ 'is-active': editor.isActive('heading', { level: 2 }) }">
+      h2
+    </button>
+    <button @click="editor.focus().heading({ level: 3 })" :class="{ 'is-active': editor.isActive('heading', { level: 3 }) }">
+      h3
+    </button>
+
+    <editor-content :editor="editor" />
+  </div>
+</template>
+
+<script>
+import { Editor } from '@tiptap/core'
+import { EditorContent } from '@tiptap/vue'
+import Document from '@tiptap/extension-document'
+import Paragraph from '@tiptap/extension-paragraph'
+import Text from '@tiptap/extension-text'
+import Heading from '@tiptap/extension-heading'
+
+export default {
+  components: {
+    EditorContent,
+  },
+
+  data() {
+    return {
+      editor: null,
+    }
+  },
+
+  mounted() {
+    this.editor = new Editor({
+      extensions: [
+        Document(),
+        Paragraph(),
+        Text(),
+        Heading({
+          levels: [1, 2, 3],
+        }),
+      ],
+      content: `
+        <h1>This is a headline level 1</h1>
+        <h2>This is a headline level 2</h2>
+        <h3>This is a headline level 3</h3>
+        <h4>This headline will be converted to a paragraph, because it's not defined in the levels option.</h4>
+      `,
+    })
+
+    window.editor = this.editor
+  },
+
+  beforeDestroy() {
+    this.editor.destroy()
+  }
+}
+</script>
\ No newline at end of file
diff --git a/docs/src/docPages/api/extensions/heading.md b/docs/src/docPages/api/extensions/heading.md
index 7c4e1965..a4e1aae3 100644
--- a/docs/src/docPages/api/extensions/heading.md
+++ b/docs/src/docPages/api/extensions/heading.md
@@ -20,53 +20,8 @@ Enables you to use headline HTML tags in the editor.
 * `Control` + `Shift` + `5` → H5
 * `Control` + `Shift` + `6` → H6
 
+## Source Code
+[packages/extension-heading/](https://github.com/ueberdosis/tiptap-next/blob/main/packages/extension-heading/)
+
 ## Usage
-```markup
-<template>
-  <div>
-    <editor-menu-bar :editor="editor" v-slot="{ commands, isActive }">
-      <div>
-        <button type="button" :class="{ 'is-active': isActive.heading({ level: 1 }) }" @click="commands.heading({ level: 1})">
-          H1
-        </button>
-        <button type="button" :class="{ 'is-active': isActive.heading({ level: 2 }) }" @click="commands.heading({ level: 2 })">
-          H2
-        </button>
-      </div>
-    </editor-menu-bar>
-
-    <editor-content :editor="editor" />
-  </div>
-</template>
-
-<script>
-import { Editor, EditorContent, EditorMenuBar } from 'tiptap'
-import { Heading } from 'tiptap-extensions'
-
-export default {
-  components: {
-    EditorMenuBar,
-    EditorContent,
-  },
-  data() {
-    return {
-      editor: new Editor({
-        extensions: [
-          Heading({
-            levels: [1, 2],
-          }),
-        ],
-        content: `
-          <h1>This is a headline level 1</h1>
-          <h2>This is a headline level 2</h2>
-          <h3>This headline will be converted to a paragraph, because it's not defined in the levels option.</h3>
-        `,
-      }),
-    }
-  },
-  beforeDestroy() {
-    this.editor.destroy()
-  }
-}
-</script>
-```
\ No newline at end of file
+<demo name="Extensions/Heading" highlight="" />