diff --git a/.drone.yml b/.drone.yml
index 2179da8..f4cd8ac 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -109,7 +109,7 @@ steps:
         - docker pull khwezi/webapitest:latest
         - docker stop webapitest || true
         - docker rm webapitest || true
-        - docker run -d --name webapitest -e ASPNETCORE_ENVIRONMENT=Development -p 4000:8081 khwezi/webapitest:latest
+        - docker run -d --name webapitest --restart unless-stopped -e ASPNETCORE_ENVIRONMENT=Development -p 4000:8081 khwezi/webapitest:latest
 
 ---
 kind: pipeline
@@ -125,18 +125,16 @@ steps:
   - name: prod
     image: appleboy/drone-ssh
     environment:
-      SSH_USER:
+      PLUGIN_USER:
         from_secret: ssh_prod_user
-      SSH_PASS:
+      PLUGIN_PASSWORD:
         from_secret: ssh_prod_password
+      PLUGIN_HOST: ssh_prod_host
     settings:
-      host: 192.168.1.48
-      username: ${SSH_USER}
-      password: ${SSH_PASS}
       script:
         - docker stop webapi || true
         - docker rm webapi || true
-        - docker run -d --name webapi -e ASPNETCORE_ENVIRONMENT=Production -p 4001:8081 khwezi/webapitest:latest
+        - docker run -d --name webapi --restart unless-stopped -e ASPNETCORE_ENVIRONMENT=Production -p 4001:8081 khwezi/webapitest:latest
     when:
       event:
         - promote