diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 1b71389..60e1f21 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -3,7 +3,8 @@ name: Build and update image
 on:
   push:
     branches:
-      - generator-imagefix
+      - master
+      - generator*
     paths: 
       - subnet/deployment-generator/**
     tags: 
diff --git a/subnet/deployment-generator/script/gen.js b/subnet/deployment-generator/script/gen.js
index cefb99c..fd418ec 100644
--- a/subnet/deployment-generator/script/gen.js
+++ b/subnet/deployment-generator/script/gen.js
@@ -385,7 +385,7 @@ function genCommands(num_machines, network_name, network_id, num_subnet, keys){
     commands+=`  docker compose --env-file docker-compose.env --profile ${machine_name} up -d\n\n` //composeV2
   }
 
-  commands+=`\nmachine1:                deploy checkpoint smart contract\n`
+  commands+=`\nmachine1:                deploy checkpoint smart contract (please be mindful of docker image tag if you are not using 'latest' \n`
   commands+=`  docker run --env-file docker.env \\
     -v $(pwd)/generated/deployment.json:/app/generated/deployment.json \\
     --entrypoint 'bash' xinfinorg/subnet-generator:latest ./deploy_csc.sh \n`         //how to inject version other than latest??