diff --git a/jaxlib/mlir/BUILD.bazel b/jaxlib/mlir/BUILD.bazel
index 3a2c285fc..19dc8d1d7 100644
--- a/jaxlib/mlir/BUILD.bazel
+++ b/jaxlib/mlir/BUILD.bazel
@@ -44,7 +44,7 @@ symlink_inputs(
 py_library(
     name = "mlir",
     deps = [
-        "//jaxlib/mlir/_mlir_libs:_mlir_libs",
+        "//jaxlib/mlir/_mlir_libs",
     ],
 )
 
@@ -106,7 +106,7 @@ symlink_inputs(
     name = "mhlo_dialect",
     rule = py_library,
     symlinked_inputs = {"srcs": {"dialects": [
-        "@org_tensorflow//tensorflow/compiler/mlir/hlo:MhloOpsPyFiles",
+        "@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:MhloOpsPyFiles",
     ]}},
     deps = [
         ":core",
diff --git a/jaxlib/mlir/_mlir_libs/BUILD.bazel b/jaxlib/mlir/_mlir_libs/BUILD.bazel
index fc4f58670..f80f00c39 100644
--- a/jaxlib/mlir/_mlir_libs/BUILD.bazel
+++ b/jaxlib/mlir/_mlir_libs/BUILD.bazel
@@ -17,7 +17,6 @@ load(
     "py_extension",
     "windows_cc_shared_mlir_library",
 )
-
 load("//jaxlib:symlink_files.bzl", "symlink_inputs")
 
 package(
@@ -75,9 +74,9 @@ py_extension(
 py_extension(
     name = "_mlirRegisterEverything",
     srcs = [
-        "@llvm-project//mlir:lib/Bindings/Python/RegisterEverything.cpp",
-        "@llvm-project//mlir:include/mlir/Bindings/Python/PybindAdaptors.h",
         "@llvm-project//mlir:include/mlir-c/Bindings/Python/Interop.h",
+        "@llvm-project//mlir:include/mlir/Bindings/Python/PybindAdaptors.h",
+        "@llvm-project//mlir:lib/Bindings/Python/RegisterEverything.cpp",
     ],
     copts = COPTS,
     deps = [
@@ -110,7 +109,7 @@ symlink_inputs(
 py_extension(
     name = "_mlirHlo",
     srcs = [
-        "@org_tensorflow//tensorflow/compiler/mlir/hlo:python/MlirHloModule.cc",
+        "@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:python/MlirHloModule.cc",
     ],
     copts = COPTS,
     deps = [
@@ -118,7 +117,7 @@ py_extension(
         "@llvm-project//mlir:CAPIIRHeaders",
         "@llvm-project//mlir:MLIRBindingsPythonHeaders",
         "@local_config_python//:headers",
-        "@org_tensorflow//tensorflow/compiler/mlir/hlo:CAPIHeaders",
+        "@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:CAPIHeaders",
         "@pybind11",
     ],
 )
@@ -139,12 +138,12 @@ cc_library(
 cc_library(
     name = "jaxlib_mlir_capi_objects",
     deps = [
-        "@llvm-project//mlir:MLIRBindingsPythonCAPIObjects",
+        "@llvm-project//mlir:CAPIConversionObjects",
+        "@llvm-project//mlir:CAPIRegisterEverythingObjects",
         "@llvm-project//mlir:CAPISparseTensorObjects",
         "@llvm-project//mlir:CAPITransformsObjects",
-        "@llvm-project//mlir:CAPIRegisterEverythingObjects",
-        "@llvm-project//mlir:CAPIConversionObjects",
-        "@org_tensorflow//tensorflow/compiler/mlir/hlo:CAPIObjects",
+        "@llvm-project//mlir:MLIRBindingsPythonCAPIObjects",
+        "@org_tensorflow//tensorflow/compiler/xla/mlir_hlo:CAPIObjects",
     ],
 )