diff --git a/script/aft_build.sh b/script/aft_build.sh
index 473bb2059ca9ccb3a3acaa4d68874d866d547e39..5a2195f0b7b2a2f7a7360af9cd5b18880d7c9e16 100644
--- a/script/aft_build.sh
+++ b/script/aft_build.sh
@@ -115,6 +115,7 @@ CONFIG_US_ALGO_C906=$(echo $CONFIG_DEFINES | grep "CONFIG_ALG_US" | awk -F ':' '
 #echo $CONFIG_DEFINES
 if [ -n "$EXE_EXT" ]; then
     # CDK
+    CONFIG_ALG_US=$(cat $MK_SOLUTION_PATH/package.yaml | grep "CONFIG_ALG_US" | awk -F ':' '{print $2}')
     CONFIG_US_ALGO_C906=$CONFIG_ALG_US
 fi