diff --git a/jupyter-codio-extension/playbook.yaml b/jupyter-codio-extension/playbook.yaml
index ac6a6ec..1c58712 100644
--- a/jupyter-codio-extension/playbook.yaml
+++ b/jupyter-codio-extension/playbook.yaml
@@ -4,19 +4,32 @@
   become_user: root
   tasks:
 
-    - name: "Install Jupyter Notebook Codio extension"
-      command: /usr/share/codio-tools/jupyter-codio-extension/scripts/installer/jupyter-codio-extension-installer --mode=notebook
-      become: yes
-      become_user: codio
-      become_method: su
-      become_flags: -l
+    # - name: "Install Jupyter Notebook Codio extension"
+    #   command: /usr/share/codio-tools/jupyter-codio-extension/scripts/installer/jupyter-codio-extension-installer --mode=notebook
+    #   become: yes
+    #   become_user: codio
+    #   become_method: su
+    #   become_flags: -l
+
+    # - name: "Install Jupyter Lab Codio extension"
+    #   command: /usr/share/codio-tools/jupyter-codio-extension/scripts/installer/jupyter-codio-extension-installer
+    #   become: yes
+    #   become_user: codio
+    #   become_method: su
+    #   become_flags: -l
 
-    - name: "Install Jupyter Lab Codio extension"
-      command: /usr/share/codio-tools/jupyter-codio-extension/scripts/installer/jupyter-codio-extension-installer
+    - name: Installing jupyter-collaboration-codio
+      pip:
+        name: jupyterlab-v4-codio-ext
+        state: present
+        version: 0.1.17
       become: yes
       become_user: codio
       become_method: su
       become_flags: -l
+      register: aptout
+
+    - debug: var=aptout
 
     - name: "Restart Jupyter"
       systemd:
diff --git a/jupyter-collaboration/playbook.yaml b/jupyter-collaboration/playbook.yaml
index 30d2dbf..88f67af 100644
--- a/jupyter-collaboration/playbook.yaml
+++ b/jupyter-collaboration/playbook.yaml
@@ -8,7 +8,7 @@
     - name: Installing jupyter-collaboration-codio
       pip:
         name: jupyter-collaboration-codio
-        extra_args: -i https://test.pypi.org/simple/
+        extra_args: -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/
         state: present
         version: 4.1.2rc2
       become: yes