diff --git a/tools/mlir-hlo-opt/mlir-hlo-opt.cpp b/tools/mlir-hlo-opt/mlir-hlo-opt.cpp
index ed96dd5..8c4ccb8 100644
--- a/tools/mlir-hlo-opt/mlir-hlo-opt.cpp
+++ b/tools/mlir-hlo-opt/mlir-hlo-opt.cpp
@@ -34,6 +34,6 @@ int main(int argc, char **argv) {
   registry.insert<mlir::lmhlo::LmhloDialect>();
   registry.insert<mlir::lmhlo_gpu::LmhloGpuDialect>();
 
-  return failed(
-      mlir::MlirOptMain(argc, argv, "MLIR HLO pass driver\n", registry));
+  return failed(mlir::MlirOptMain(argc, argv, "MLIR HLO pass driver\n",
+                                  registry, /*preloadDialectsInContext=*/true));
 }