diff --git a/src/opt/apps/@PKG_NAME@/files/bin/ace-init b/src/opt/apps/@PKG_NAME@/files/bin/ace-init
index 893ae7c..a62dd55 100755
--- a/src/opt/apps/@PKG_NAME@/files/bin/ace-init
+++ b/src/opt/apps/@PKG_NAME@/files/bin/ace-init
@@ -25,7 +25,7 @@ HERE="$(dirname $(realpath $0))"
 function get_current_user() {
     # 优先通过 who 命令获取用户
     local user
-    user=$(who | awk '{print $1}' | head -n 1 2>/dev/null)
+    user=$(get_current_user 2>/dev/null)
 
     # 如果 who 无输出，则通过 loginctl 获取
     if [[ -z "$user" ]]; then
@@ -86,8 +86,8 @@ chmod 777 $chrootEnvPath/etc/localtime
 bookworm-run cp /host/etc/locale.gen /etc/locale.gen && locale-gen
 bookworm-run touch /finish.flag
 bookworm-run apt clean
-bookworm-run chown -R $(who | awk '{print $1}' | head -n 1)  /usr/lib/locale/
-sudo -u $(who | awk '{print $1}' | head -n 1) bwrap --dev-bind $chrootEnvPath/ / \
+bookworm-run chown -R $(get_current_user)  /usr/lib/locale/
+sudo -u $(get_current_user) bwrap --dev-bind $chrootEnvPath/ / \
   --setenv PULSE_SERVER /run/user/$uid/pulse/native \
   --setenv PATH /amber-ce-tools/bin-override:$PATH \
   --setenv IS_ACE_ENV "1" \