diff --git a/compiler/include/concretelang/Conversion/Tools.h b/compiler/include/concretelang/Conversion/Tools.h
index 949a0eb68..46ab11d98 100644
--- a/compiler/include/concretelang/Conversion/Tools.h
+++ b/compiler/include/concretelang/Conversion/Tools.h
@@ -6,6 +6,6 @@
 #include "mlir/IR/PatternMatch.h"
 
 mlir::LogicalResult insertForwardDeclaration(mlir::Operation *op,
-                                             mlir::RewriterBase &rewriter,
+                                             mlir::OpBuilder &rewriter,
                                              llvm::StringRef funcName,
                                              mlir::FunctionType funcType);
diff --git a/compiler/lib/Conversion/Tools.cpp b/compiler/lib/Conversion/Tools.cpp
index 5aa64fc2c..6b0c0f5d9 100644
--- a/compiler/lib/Conversion/Tools.cpp
+++ b/compiler/lib/Conversion/Tools.cpp
@@ -8,7 +8,7 @@
 #include "concretelang/Conversion/Tools.h"
 
 mlir::LogicalResult insertForwardDeclaration(mlir::Operation *op,
-                                             mlir::RewriterBase &rewriter,
+                                             mlir::OpBuilder &rewriter,
                                              llvm::StringRef funcName,
                                              mlir::FunctionType funcType) {
   // Looking for the `funcName` Operation