diff --git a/workshop/comfy_new.yaml b/workshop/comfy_new.yaml
index 3cab0561..3612e5a9 100644
--- a/workshop/comfy_new.yaml
+++ b/workshop/comfy_new.yaml
@@ -213,6 +213,7 @@ Resources:
           echo "PROCESS_NUMBER=${UserNumber}" >> /etc/environment
           echo "ESD_VERSION=${EsdVersion}" >> /etc/environment
           echo "SERVICE_TYPE=comfy" >> /etc/environment
+          echo "ON_EC2=true" >> /etc/environment
           
           source /etc/environment