diff --git a/entrypoint.sh b/entrypoint.sh
index a09c284..6076438 100644
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -13,14 +13,14 @@ fi
 # Otherwise the current container UID may not exist in the passwd database.
 eval "$(fixuid -q)"
 
-if [ "${HOME_USER-}" ]; then
-  USER="$HOME_USER"
-  if [ "$HOME_USER" != "$(whoami)" ]; then
-    echo "$HOME_USER ALL=(ALL) NOPASSWD:ALL" | sudo tee -a /etc/sudoers.d/nopasswd > /dev/null
+if [ "${DOCKER_USER-}" ]; then
+  USER="$DOCKER_USER"
+  if [ "$DOCKER_USER" != "$(whoami)" ]; then
+    echo "$DOCKER_USER ALL=(ALL) NOPASSWD:ALL" | sudo tee -a /etc/sudoers.d/nopasswd > /dev/null
     # Unfortunately we cannot change $HOME as we cannot move any bind mounts
     # nor can we bind mount $HOME into a new home as that requires a privileged container.
-    sudo usermod --login "$HOME_USER" vscode
-    sudo groupmod -n "$HOME_USER" vscode
+    sudo usermod --login "$DOCKER_USER" vscode
+    sudo groupmod -n "$DOCKER_USER" vscode
 
     sudo sed -i "/vscode/d" /etc/sudoers.d/nopasswd
   fi
diff --git a/version.txt b/version.txt
index 359a5b9..10bf840 100644
--- a/version.txt
+++ b/version.txt
@@ -1 +1 @@
-2.0.0
\ No newline at end of file
+2.0.1
\ No newline at end of file