4 files changed, 4 insertions, 4 deletions
diff --git a/mlir/test/lib/Transforms/TestConstantFold.cpp b/mlir/test/lib/Transforms/TestConstantFold.cpp
index b1c895257c3..15ecaabb149 100644
--- a/mlir/test/lib/Transforms/TestConstantFold.cpp
+++ b/mlir/test/lib/Transforms/TestConstantFold.cpp
@@ -74,7 +74,7 @@ void TestConstantFold::runOnFunction() {
 }
 
 /// Creates a constant folding pass.
-std::unique_ptr<FunctionPassBase> mlir::createTestConstantFoldPass() {
+std::unique_ptr<OpPassBase<FuncOp>> mlir::createTestConstantFoldPass() {
   return std::make_unique<TestConstantFold>();
 }
 
diff --git a/mlir/test/lib/Transforms/TestLoopFusion.cpp b/mlir/test/lib/Transforms/TestLoopFusion.cpp
index 604b42817e2..026a897fa8d 100644
--- a/mlir/test/lib/Transforms/TestLoopFusion.cpp
+++ b/mlir/test/lib/Transforms/TestLoopFusion.cpp
@@ -58,7 +58,7 @@ struct TestLoopFusion : public FunctionPass<TestLoopFusion> {
 
 } // end anonymous namespace
 
-std::unique_ptr<FunctionPassBase> mlir::createTestLoopFusionPass() {
+std::unique_ptr<OpPassBase<FuncOp>> mlir::createTestLoopFusionPass() {
   return std::make_unique<TestLoopFusion>();
 }
 
diff --git a/mlir/test/lib/Transforms/TestLoopParametricTiling.cpp b/mlir/test/lib/Transforms/TestLoopParametricTiling.cpp
index 6dc0bfde371..bce1e08402d 100644
--- a/mlir/test/lib/Transforms/TestLoopParametricTiling.cpp
+++ b/mlir/test/lib/Transforms/TestLoopParametricTiling.cpp
@@ -55,7 +55,7 @@ public:
 };
 } // end namespace
 
-std::unique_ptr<FunctionPassBase>
+std::unique_ptr<OpPassBase<FuncOp>>
 mlir::createSimpleParametricTilingPass(ArrayRef<int64_t> outerLoopSizes) {
   return std::make_unique<SimpleParametricLoopTilingPass>(outerLoopSizes);
 }
diff --git a/mlir/test/lib/Transforms/TestVectorizationUtils.cpp b/mlir/test/lib/Transforms/TestVectorizationUtils.cpp
index 3f00eb01e11..4fdb66071bf 100644
--- a/mlir/test/lib/Transforms/TestVectorizationUtils.cpp
+++ b/mlir/test/lib/Transforms/TestVectorizationUtils.cpp
@@ -290,7 +290,7 @@ void VectorizerTestPass::runOnFunction() {
   }
 }
 
-std::unique_ptr<FunctionPassBase> mlir::createVectorizerTestPass() {
+std::unique_ptr<OpPassBase<FuncOp>> mlir::createVectorizerTestPass() {
   return std::make_unique<VectorizerTestPass>();
 }