diff --git a/install.sh b/install.sh
index a8b28a2..a7bbc4f 100755
--- a/install.sh
+++ b/install.sh
@@ -14,11 +14,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