diff --git a/install.sh b/install.sh
index 0e87d85..f5640c1 100755
--- a/install.sh
+++ b/install.sh
@@ -15,11 +15,13 @@ echo "Installing dotfiles..."
 if [ "${CODER_TEMPLATE_TYPE}" == "Devcontainer" ]; then
   installers/install-nix.sh
   installers/install-nix-packages-flake.sh
+  . ~/.nix-profile/etc/profile.d/nix.sh
 elif [ "${CODER_OS_FAMILY:-}" == "Ubuntu" ]; then
   echo "Detected OS as Ubuntu, using apt";
   sudo apt-get update;
   sudo apt-get install -y stow;
   installers/install-nix.sh
+  . ~/.nix-profile/etc/profile.d/nix.sh
 elif [ -f /etc/os-release ]; then
   echo "Idenitfying OS to install GNU stow..."
   if grep -q -i debian /etc/os-release; then