diff --git a/docs/package.json b/docs/package.json
index 3581fd3d..2ad3f3ae 100644
--- a/docs/package.json
+++ b/docs/package.json
@@ -21,7 +21,9 @@
     "react-dom": "^16.13.1",
     "remark-container": "^0.1.2",
     "typescript": "^4.0.3",
-    "vue-github-button": "^1.1.2"
+    "vue-github-button": "^1.1.2",
+    "y-webrtc": "^10.1.6",
+    "yjs": "^13.3.2"
   },
   "devDependencies": {
     "@babel/plugin-proposal-optional-chaining": "^7.11.0",
diff --git a/packages/extension-collaboration-cursor/package.json b/packages/extension-collaboration-cursor/package.json
index 3035a014..a4410c04 100644
--- a/packages/extension-collaboration-cursor/package.json
+++ b/packages/extension-collaboration-cursor/package.json
@@ -15,8 +15,6 @@
     "@tiptap/core": "2.x"
   },
   "dependencies": {
-    "y-prosemirror": "^0.3.7",
-    "y-webrtc": "^10.1.6",
-    "yjs": "^13.3.2"
+    "y-prosemirror": "^0.3.7"
   }
 }
diff --git a/packages/extension-collaboration/index.ts b/packages/extension-collaboration/index.ts
index 139ee464..e734b370 100644
--- a/packages/extension-collaboration/index.ts
+++ b/packages/extension-collaboration/index.ts
@@ -2,7 +2,6 @@ import { Extension } from '@tiptap/core'
 import {
   redo, undo, ySyncPlugin, yUndoPlugin,
 } from 'y-prosemirror'
-import { keymap } from 'prosemirror-keymap'
 
 export interface CollaborationOptions {
   provider: any,
@@ -18,10 +17,12 @@ export default new Extension<CollaborationOptions>()
   .plugins(({ options }) => [
     ySyncPlugin(options.type),
     yUndoPlugin(),
-    keymap({
+  ])
+  .keys(() => {
+    return {
       'Mod-z': undo,
       'Mod-y': redo,
       'Mod-Shift-z': redo,
-    }),
-  ])
+    }
+  })
   .create()
diff --git a/packages/extension-collaboration/package.json b/packages/extension-collaboration/package.json
index 4fa5d67b..ebb475d0 100644
--- a/packages/extension-collaboration/package.json
+++ b/packages/extension-collaboration/package.json
@@ -15,8 +15,6 @@
     "@tiptap/core": "2.x"
   },
   "dependencies": {
-    "y-prosemirror": "^0.3.7",
-    "y-webrtc": "^10.1.6",
-    "yjs": "^13.3.2"
+    "y-prosemirror": "^0.3.7"
   }
 }