diff --git a/setup b/setup
index faa0244..cfa14d0 100755
--- a/setup
+++ b/setup
@@ -20,6 +20,9 @@ echo "[+] Creating symlinks..."
 
 if [ ! -f "$PWD/.zshenv" ]; then
     cp "$PWD/.zshenv.template" "$PWD/.zshenv"
+
+    SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+    sed -i "s|^export DOTFILES_INSTALL_DIR=.*|export DOTFILES_INSTALL_DIR=${SCRIPT_DIR}|" .zshenv
 fi
 
 stow .