diff --git a/docs/src/demos/Examples/Basic/index.vue b/docs/src/demos/Examples/Basic/index.vue
index 70b52b13..76450865 100644
--- a/docs/src/demos/Examples/Basic/index.vue
+++ b/docs/src/demos/Examples/Basic/index.vue
@@ -10,7 +10,7 @@
       <button @click="editor.chain().focus().toggleStrike().run()" :class="{ 'is-active': editor.isActive('strike') }">
         strike
       </button>
-      <button @click="editor.chain().focus().code().run()" :class="{ 'is-active': editor.isActive('code') }">
+      <button @click="editor.chain().focus().toggleCode().run()" :class="{ 'is-active': editor.isActive('code') }">
         code
       </button>
       <button @click="editor.chain().focus().removeMarks().run()">
diff --git a/docs/src/demos/Examples/CollaborativeEditing/index.vue b/docs/src/demos/Examples/CollaborativeEditing/index.vue
index 17d156ab..e14f98e0 100644
--- a/docs/src/demos/Examples/CollaborativeEditing/index.vue
+++ b/docs/src/demos/Examples/CollaborativeEditing/index.vue
@@ -10,7 +10,7 @@
       <button @click="editor.chain().focus().toggleStrike().run()" :class="{ 'is-active': editor.isActive('strike') }">
         strike
       </button>
-      <button @click="editor.chain().focus().code().run()" :class="{ 'is-active': editor.isActive('code') }">
+      <button @click="editor.chain().focus().toggleCode().run()" :class="{ 'is-active': editor.isActive('code') }">
         code
       </button>
       <button @click="editor.chain().focus().removeMarks().run()">
diff --git a/docs/src/demos/Examples/CollaborativeEditingWs/index.vue b/docs/src/demos/Examples/CollaborativeEditingWs/index.vue
index 38ee356f..2bc267ba 100644
--- a/docs/src/demos/Examples/CollaborativeEditingWs/index.vue
+++ b/docs/src/demos/Examples/CollaborativeEditingWs/index.vue
@@ -10,7 +10,7 @@
       <button @click="editor.chain().focus().toggleStrike().run()" :class="{ 'is-active': editor.isActive('strike') }">
         strike
       </button>
-      <button @click="editor.chain().focus().code().run()" :class="{ 'is-active': editor.isActive('code') }">
+      <button @click="editor.chain().focus().toggleCode().run()" :class="{ 'is-active': editor.isActive('code') }">
         code
       </button>
       <button @click="editor.chain().focus().removeMarks().run()">
diff --git a/docs/src/demos/Marks/Code/index.vue b/docs/src/demos/Marks/Code/index.vue
index ac874df5..365f8208 100644
--- a/docs/src/demos/Marks/Code/index.vue
+++ b/docs/src/demos/Marks/Code/index.vue
@@ -1,6 +1,6 @@
 <template>
   <div v-if="editor">
-    <button @click="editor.chain().focus().code().run()" :class="{ 'is-active': editor.isActive('code') }">
+    <button @click="editor.chain().focus().toggleCode().run()" :class="{ 'is-active': editor.isActive('code') }">
       code
     </button>
 
diff --git a/packages/extension-code/src/index.ts b/packages/extension-code/src/index.ts
index 5eb5909e..c68b9ff0 100644
--- a/packages/extension-code/src/index.ts
+++ b/packages/extension-code/src/index.ts
@@ -35,18 +35,30 @@ const Code = Mark.create({
 
   addCommands() {
     return {
+      /**
+       * Set a code mark
+       */
+      setCode: (): Command => ({ commands }) => {
+        return commands.addMark('code')
+      },
       /**
        * Toggle inline code
        */
-      code: (): Command => ({ commands }) => {
+      toggleCode: (): Command => ({ commands }) => {
         return commands.toggleMark('code')
       },
+      /**
+       * Set a code mark
+       */
+      unsetCode: (): Command => ({ commands }) => {
+        return commands.addMark('code')
+      },
     }
   },
 
   addKeyboardShortcuts() {
     return {
-      'Mod-`': () => this.editor.commands.code(),
+      'Mod-`': () => this.editor.commands.toggleCode(),
     }
   },