1 files changed, 3 insertions, 3 deletions
diff --git a/mlir/lib/Transforms/ViewRegionGraph.cpp b/mlir/lib/Transforms/ViewRegionGraph.cpp
index 5a0e8e5ea99..91ac397200a 100644
--- a/mlir/lib/Transforms/ViewRegionGraph.cpp
+++ b/mlir/lib/Transforms/ViewRegionGraph.cpp
@@ -85,9 +85,9 @@ private:
 };
 } // namespace
 
-FunctionPassBase *mlir::createPrintCFGGraphPass(llvm::raw_ostream &os,
-                                                bool shortNames,
-                                                const llvm::Twine &title) {
+OpPassBase<FuncOp> *mlir::createPrintCFGGraphPass(llvm::raw_ostream &os,
+                                                  bool shortNames,
+                                                  const llvm::Twine &title) {
   return new PrintCFGPass(os, shortNames, title);
 }