diff --git a/templates/portainer-agent-stack.yml.j2 b/templates/portainer-agent-stack.yml.j2
index d8ea26f..a348102 100644
--- a/templates/portainer-agent-stack.yml.j2
+++ b/templates/portainer-agent-stack.yml.j2
@@ -23,9 +23,6 @@ services:
     #ports:
     #  - "{{ docker_swarm_portainer_http_port }}:9000"
     #  - "{{ docker_swarm_portainer_service_port }}:8000"
-    expose:
-      - "9000"
-      - "8000"
     volumes:
       - portainer_data:/data
     networks:
@@ -35,7 +32,7 @@ services:
       replicas: 1
       endpoint_mode: dnsrr
       restart_policy:
-        condition: unless-stopped
+        condition: on-failure
         delay: 5s
         max_attempts: 3
         window: 120s