diff --git a/src/PIM/Pass/MessagePass.cpp b/src/PIM/Pass/MessagePass.cpp
index 01224fd..3a897a5 100644
--- a/src/PIM/Pass/MessagePass.cpp
+++ b/src/PIM/Pass/MessagePass.cpp
@@ -1,7 +1,7 @@
 #include "mlir/IR/BuiltinOps.h"
 #include "mlir/Pass/Pass.h"
 
-#include "llvm/Support/raw_ostream.h"
+#include "src/Compiler/CompilerUtils.hpp"
 
 using namespace mlir;
 
@@ -18,10 +18,7 @@ struct MessagePass : PassWrapper<MessagePass, OperationPass<ModuleOp>> {
   : message(message) {}
   MessagePass(const MessagePass& pass) {}
 
-  void runOnOperation() final {
-    llvm::outs() << message << "\n";
-    llvm::outs().flush();
-  }
+  void runOnOperation() final { showCompilePhase(message); }
 
 private:
   std::string message;