diff --git a/workstation.yml b/workstation.yml
index 5217dd7144de4f28362416402cce501b85ac657a..038b9f7476585c48bf5ed54973a4c0891783da13 100644
--- a/workstation.yml
+++ b/workstation.yml
@@ -25,7 +25,7 @@
   - name: APT install
     with_items:
     - network-manager
-    - docker.io
+    - docker-ce
     - ufw
     - libdvdcss2
     - vagrant
@@ -84,11 +84,12 @@
     - Restart Docker
 
   - name: Configure Docker graph directory
-    lineinfile:
-      dest: /etc/default/docker
-      line: 'DOCKER_OPTS="--graph {{ docker_graph_directory }}"'
-      regexp: '^DOCKER_OPTS='
-      state: present
+    copy:
+        content: "{{ {'graph': docker_graph_directory}|to_json }}"
+        dest: /etc/docker/daemon.json
+        owner: root
+        group: root
+        mode: 0o0644
     notify:
     - Restart Docker