diff --git a/playbooks/apps/files/test_headers/docker-compose.yaml b/playbooks/apps/files/test_headers/docker-compose.yaml
index 2f8d517..65c2e1b 100644
--- a/playbooks/apps/files/test_headers/docker-compose.yaml
+++ b/playbooks/apps/files/test_headers/docker-compose.yaml
@@ -6,3 +6,5 @@ services:
       - ./:/app
     ports:
       - "9008:8080"
+    init: true
+    tty: true
diff --git a/playbooks/apps/test_headers.yaml b/playbooks/apps/test_headers.yaml
index 60e2e81..de95069 100644
--- a/playbooks/apps/test_headers.yaml
+++ b/playbooks/apps/test_headers.yaml
@@ -3,6 +3,11 @@
   hosts: servers
   gather_facts: false
   tasks:
+    - name: Create dir
+      ansible.builtin.file:
+        path: "{{ dir }}"
+        state: directory
+        mode: "0775"
     - name: Write app.py
       ansible.builtin.copy:
         src: files/test_headers/app.py
@@ -10,6 +15,8 @@
         mode: "0644"
         owner: gaugendre
         group: gaugendre
+      notify:
+        - Restart service
     - name: Write docker-compose.yaml
       ansible.builtin.copy:
         src: files/test_headers/docker-compose.yaml
@@ -21,5 +28,10 @@
       community.docker.docker_compose_v2:
         project_src: "{{ dir }}"
         state: present
+  handlers:
+    - name: Restart service
+      community.docker.docker_compose_v2:
+        project_src: "{{ dir }}"
+        state: restarted
   vars:
     dir: /mnt/data/test_headers