diff --git a/templates/docker-compose.yml.j2 b/templates/docker-compose.yml.j2
index d16a603..200e967 100644
--- a/templates/docker-compose.yml.j2
+++ b/templates/docker-compose.yml.j2
@@ -192,8 +192,6 @@ services:
 {% for device_id in device.device_ids %}
                 - {{ device_id }}
 {% endfor %}
-{% endif %}
-{% endfor %}
 {% else %}
     deploy:
         resources:
@@ -202,6 +200,8 @@ services:
             - count: {{ container.deploy.resources.reservations.devices[0].driver.count | default("1") }}
                 capabilities: [gpu]
 {% endif %}
+{% endfor %}
+{% endif %}
 {% if container.restart is defined %}
     restart: {{ container.restart }}
 {% endif %}