diff --git a/aztec-up/bin/aztec-wallet b/aztec-up/bin/aztec-wallet
index d67e5f8c99f..1928b4a5e80 100755
--- a/aztec-up/bin/aztec-wallet
+++ b/aztec-up/bin/aztec-wallet
@@ -3,11 +3,11 @@ set -euo pipefail
 
 export SKIP_PORT_ASSIGNMENT=1
 export WALLET_DATA_DIRECTORY=$(dirname $0)/wallet-data
-export ENV_VARS_TO_INJECT="WALLET_DATA_DIRECTORY SSH_AUTH_SOCK"
 export BB_BINARY_PATH=$HOME/.bb/bb
-export BB_WORKING_DIRECTORY=$HOME/.bb-workdir
+export BB_WORKING_DIRECTORY=$(dirname $0)/bb-workdir
+export ENV_VARS_TO_INJECT="WALLET_DATA_DIRECTORY SSH_AUTH_SOCK PXE_PROVER_ENABLED BB_BINARY_PATH BB_WORKING_DIRECTORY"
 
 mkdir -p $BB_WORKING_DIRECTORY
 mkdir -p $WALLET_DATA_DIRECTORY
 
-$(dirname $0)/.aztec-run aztecprotocol/cli-wallet $@
\ No newline at end of file
+./.aztec-run aztecprotocol/cli-wallet $@
\ No newline at end of file