diff --git a/include/tfrt/bef_converter/mlir_to_bef_translate.h b/include/tfrt/bef_converter/mlir_to_bef_translate.h
index bf6334fc195..7d0781ed80e 100644
--- a/include/tfrt/bef_converter/mlir_to_bef_translate.h
+++ b/include/tfrt/bef_converter/mlir_to_bef_translate.h
@@ -23,12 +23,11 @@
 
 namespace mlir {
 class ModuleOp;
-struct LogicalResult;
 }  // namespace mlir
 
 namespace tfrt {
 
-mlir::LogicalResult MLIRToBEFTranslate(mlir::ModuleOp module,
+llvm::LogicalResult MLIRToBEFTranslate(mlir::ModuleOp module,
                                        llvm::raw_ostream& output);
 
 }
diff --git a/include/tfrt/support/forward_decls.h b/include/tfrt/support/forward_decls.h
index c3a42504d96..3b0ca21146b 100644
--- a/include/tfrt/support/forward_decls.h
+++ b/include/tfrt/support/forward_decls.h
@@ -25,6 +25,7 @@
 
 #include "llvm/Support/Casting.h"
 #include "llvm/Support/ErrorHandling.h"
+#include "llvm/Support/LogicalResult.h"
 
 // We don't forward declare:
 //   DenseMap, SmallVector, StringMap, StringSet
@@ -32,6 +33,7 @@
 namespace llvm {
 
 class raw_ostream;
+struct LogicalResult;
 
 template <typename T>
 class ArrayRef;