diff --git a/packages/tiptap/src/Utils/ComponentView.js b/packages/tiptap/src/Utils/ComponentView.js
index c72d52b18..64c620097 100644
--- a/packages/tiptap/src/Utils/ComponentView.js
+++ b/packages/tiptap/src/Utils/ComponentView.js
@@ -41,6 +41,10 @@ export default class ComponentView {
       updateAttrs: attrs => this.updateAttrs(attrs),
     }
 
+    if (typeof this.extension.setSelection === 'function') {
+      this.setSelection = this.extension.setSelection
+    }
+
     this.vm = new Component({
       parent: this.parent,
       propsData: props,