diff --git a/docker_start.sh b/docker_start.sh
index 129970db..2a1af301 100755
--- a/docker_start.sh
+++ b/docker_start.sh
@@ -82,7 +82,8 @@ WORKDIR /home/ubuntu/ComfyUI"
   START_HANDLER="#!/bin/bash
 set -euxo pipefail
 
-WORKFLOW_NAME=\$(cat $CONTAINER_PATH/$PROGRAM_NAME)
+export ACCOUNT_ID=$(aws sts get-caller-identity --query "Account" --output text)
+export WORKFLOW_NAME=\$(cat $CONTAINER_PATH/$PROGRAM_NAME)
 
 if [ \"\$WORKFLOW_NAME\" = \"default\" ]; then
   BASE_IMAGE=$PUBLIC_BASE_IMAGE