diff --git a/integrations/pi-zai-usage/scripts/zai-usage.sh b/integrations/pi-zai-usage/scripts/zai-usage.sh
index 1258e67..258fdb0 100755
--- a/integrations/pi-zai-usage/scripts/zai-usage.sh
+++ b/integrations/pi-zai-usage/scripts/zai-usage.sh
@@ -5,7 +5,7 @@ set -euo pipefail
 
 HOURS="${1:-24}"
 AUTH_JSON="$HOME/.pi/agent/auth.json"
-KEY="$(python3 -c "import json;print(json.load(open('$AUTH_JSON'))['zai']['key'])")"
+export KEY="$(python3 -c "import json;print(json.load(open('$AUTH_JSON'))['zai']['key'])")"
 BASE="https://api.z.ai/api/monitor/usage"
 
 NOW="$(date -u +'%Y-%m-%d %H:%M:%S')"