diff --git a/src/ebmc/neural_liveness.cpp b/src/ebmc/neural_liveness.cpp
index 3c897a570..07574a413 100644
--- a/src/ebmc/neural_liveness.cpp
+++ b/src/ebmc/neural_liveness.cpp
@@ -318,7 +318,7 @@ tvt neural_livenesst::verify(
 
 int do_neural_liveness(
   const cmdlinet &cmdline,
-  ui_message_handlert &ui_message_handler)
+  message_handlert &message_handler)
 {
-  return neural_livenesst(cmdline, ui_message_handler)();
+  return neural_livenesst(cmdline, message_handler)();
 }
diff --git a/src/ebmc/neural_liveness.h b/src/ebmc/neural_liveness.h
index 159f38208..34766b617 100644
--- a/src/ebmc/neural_liveness.h
+++ b/src/ebmc/neural_liveness.h
@@ -10,8 +10,8 @@ Author: Daniel Kroening, dkr@amazon.com
 #define EBMC_NEURAL_LIVENESS_H
 
 #include <util/cmdline.h>
-#include <util/ui_message.h>
+#include <util/message.h>
 
-int do_neural_liveness(const cmdlinet &, ui_message_handlert &);
+int do_neural_liveness(const cmdlinet &, message_handlert &);
 
 #endif // EBMC_NEURAL_LIVENESS_H