diff --git a/external/torch-mlir/build_tools/update_torch_ods.sh b/external/torch-mlir/build_tools/update_torch_ods.sh
index 25ae1a443f5a..8e1de960282e 100755
--- a/external/torch-mlir/build_tools/update_torch_ods.sh
+++ b/external/torch-mlir/build_tools/update_torch_ods.sh
@@ -2,10 +2,10 @@
 # Updates auto-generated ODS files for the `torch` dialect.
 set -e
 
-src_dir="$(realpath $(dirname $0)/..)"
+src_dir="$(realpath $(dirname $0)/../../..)"
 build_dir="$(realpath "${TORCH_MLIR_BUILD_DIR:-$src_dir/build}")"
-torch_ir_dir="${src_dir}/include/torch-mlir/Dialect/Torch/IR"
-python_packages_dir="${build_dir}/tools/torch-mlir/python_packages"
+torch_ir_dir="${src_dir}/external/torch-mlir/include/torch-mlir/Dialect/Torch/IR"
+python_packages_dir="${build_dir}/python_packages"
 
 #ninja -C "${build_dir}"
 PYTHONPATH="${python_packages_dir}/torch_mlir" python \