diff --git a/mlir-tensorrt/BUILD b/mlir-tensorrt/BUILD
index fe5e03ee9..2967904ac 100644
--- a/mlir-tensorrt/BUILD
+++ b/mlir-tensorrt/BUILD
@@ -697,7 +697,7 @@ cc_binary(
     copts = [
         "-DMLIR_TRT_TARGET_TENSORRT",
     ],
-    linkopts = ["-Wl,-rpath,./bazel-mlir-tensorrt/external/tensorrt10_x86/targets/x86_64-linux-gnu/lib"],
+    linkopts = ["-Wl,-rpath,/opt/src/mlir-tensorrt/bazel-mlir-tensorrt/external/tensorrt10_x86/targets/x86_64-linux-gnu/lib"],
     deps = [
         ":TensorRTRegistration",
         "@llvm-project//mlir:MlirOptLib",
diff --git a/mlir-tensorrt/build_tools/bazel/Dockerfile b/mlir-tensorrt/build_tools/bazel/Dockerfile
index 2ab9a78ba..01cd872d3 100644
--- a/mlir-tensorrt/build_tools/bazel/Dockerfile
+++ b/mlir-tensorrt/build_tools/bazel/Dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE
+ARG BASE_IMAGE=nvcr.io/nvidia/cuda:12.5.1-cudnn-devel-ubuntu22.04
 FROM ${BASE_IMAGE}
 
 # Specify user IDs