diff --git a/tools/updater.cpp b/tools/updater.cpp
index ca9dadb..a5007b2 100644
--- a/tools/updater.cpp
+++ b/tools/updater.cpp
@@ -107,7 +107,8 @@
 }
 
 /* Poll an open verification session.  Handling closing the session is not yet
- * owned by this method. */
+ * owned by this method.
+ */
 bool pollStatus(std::uint16_t session, ipmiblob::BlobInterface* blob)
 {
     using namespace std::chrono_literals;
@@ -230,7 +231,7 @@
     return (success == true);
 }
 
-void updaterMain(UpdateHandler* updater, const std::string& imagePath,
+void updaterMain(UpdateHandlerInterface* updater, const std::string& imagePath,
                  const std::string& signaturePath)
 {
     /* TODO(venture): Add optional parameter to specify the flash type, default