diff --git a/finetune/Dockerfile.update b/finetune/Dockerfile.update
index 52eb7a1..9d5f049 100644
--- a/finetune/Dockerfile.update
+++ b/finetune/Dockerfile.update
@@ -19,7 +19,7 @@ SHELL ["/bin/bash", "-c"]
 RUN <<EOT
 #!/bin/bash
 apt-get update
-apt-get install -y wget curl htop jq vim bash libaio-dev build-essential openssh-server python3 python3-pip bzip2
+apt-get install -y bash-completion wget curl htop jq vim bash libaio-dev build-essential openssh-server python3 python3-pip bzip2
 apt-get install -y --no-install-recommends software-properties-common build-essential autotools-dev nfs-common pdsh cmake g++ gcc curl wget vim tmux emacs less unzip htop iftop iotop ca-certificates openssh-client openssh-server rsync iputils-ping net-tools sudo llvm-dev re2c
 add-apt-repository ppa:git-core/ppa -y
 apt-get install -y git libnuma-dev wget
@@ -118,7 +118,7 @@ RUN <<EOT
 source /opt/conda/etc/profile.d/conda.sh
 conda activate ${CONDA_ENV_NAME}
 git clone https://github.com/NVIDIA/apex ${STAGE_DIR}/apex
-cd apex
+cd ${STAGE_DIR}/apex
 # if pip >= 23.1 (ref: https://pip.pypa.io/en/stable/news/#v23-1) which supports multiple `--config-settings` with the same key... 
 MAX_JOBS=1 python3 -m pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./
 python -c "import apex.amp; print('Apex is installed and the amp module is available.')"