diff --git a/jupyter-collaboration/playbook.yaml b/jupyter-collaboration/playbook.yaml
index 7a37366..30d2dbf 100644
--- a/jupyter-collaboration/playbook.yaml
+++ b/jupyter-collaboration/playbook.yaml
@@ -5,9 +5,12 @@
 
     - apt: update_cache=yes
 
-    - name: Installing notebook and nbgrader
+    - name: Installing jupyter-collaboration-codio
       pip:
-        name: jupyter-collaboration
+        name: jupyter-collaboration-codio
+        extra_args: -i https://test.pypi.org/simple/
+        state: present
+        version: 4.1.2rc2
       become: yes
       become_user: codio
       become_method: su
@@ -16,56 +19,6 @@
 
     - debug: var=aptout
 
-    - name: Remove jupyter-server-ydoc
-      pip:
-        name: jupyter-server-ydoc
-        state: absent
-      become: yes
-      become_user: codio
-      become_method: su
-      become_flags: -l
-      register: aptout1
-
-    - debug: var=aptout1
-
-    - name: Install jupyter-server-ydoc-codio
-      pip:
-        name: jupyter-server-ydoc-codio
-        state: present
-        version: 2.1.1
-      become: yes
-      become_user: codio
-      become_method: su
-      become_flags: -l
-      register: aptout2
-
-    - debug: var=aptout2
-
-    - name: Remove jupyter-docprovider
-      pip:
-        name: jupyter-docprovider
-        state: absent
-      become: yes
-      become_user: codio
-      become_method: su
-      become_flags: -l
-      register: aptoutdocprovider
-
-    - debug: var=aptoutdocprovider
-
-    - name: Install jupyter-docprovider-codio
-      pip:
-        name: jupyter-docprovider-codio
-        state: present
-        version: 2.1.1
-      become: yes
-      become_user: codio
-      become_method: su
-      become_flags: -l
-      register: aptoutdocprovider2
-
-    - debug: var=aptoutdocprovider2
-
     - name: jupyter restarted
       systemd:
         enabled: yes