diff --git a/partition/roles/ztp/templates/ztp.json.j2 b/partition/roles/ztp/templates/ztp.json.j2
index 4bfce71a..55fb5c23 100644
--- a/partition/roles/ztp/templates/ztp.json.j2
+++ b/partition/roles/ztp/templates/ztp.json.j2
@@ -5,7 +5,7 @@
         "url": "http://{{ ztp_listen_address }}:{{ ztp_port }}/ztp.sh"
       }
     },
-    "03-additional-files": {
+    "03-download": {
       "files": [
         {
           "url": {
@@ -32,10 +32,8 @@
       "clear-config": true
     },
 {% if ztp_additional_files | selectattr('name', 'equalto', 'ztp-additional-script.sh') %}
-    "05-ztp-additional-script": {
-      "plugin": {
-        "url": "http://{{ ztp_listen_address }}:{{ ztp_port }}/ztp-additional-script.sh"
-      }
+    "05-plugin": {
+      "url": "http://{{ ztp_listen_address }}:{{ ztp_port }}/ztp-additional-script.sh"
     },
 {% endif %}
     "reboot-on-success": true,