diff --git a/build_tools/write_env_file.sh b/build_tools/write_env_file.sh
index 05179c56a07c..8f3c9a59357f 100755
--- a/build_tools/write_env_file.sh
+++ b/build_tools/write_env_file.sh
@@ -13,7 +13,7 @@ portable_realpath() {
 
 td="$(portable_realpath "$(dirname "$0")"/..)"
 build_dir="$(portable_realpath "${TORCH_MLIR_BUILD_DIR:-$td/build}")"
-python_packages_dir="$build_dir/tools/torch-mlir/python_packages"
+python_packages_dir="$build_dir/python_packages"
 
 write_env_file() {
   echo "Updating $build_dir/.env file"
diff --git a/docs/development.md b/docs/development.md
index f1e72966f84d..360dff8f9df8 100644
--- a/docs/development.md
+++ b/docs/development.md
@@ -197,7 +197,7 @@ TIP: add multiple target options to stack build phases
 #### Linux and macOS
 
 ```shell
-export PYTHONPATH=`pwd`/build/tools/torch-mlir/python_packages/torch_mlir:`pwd`/test/python/fx_importer
+export PYTHONPATH=`pwd`/build/python_packages/torch_mlir:`pwd`/test/python/fx_importer
 ```
 
 #### Windows PowerShell