diff --git a/tools/test/tools_updater_unittest.cpp b/tools/test/tools_updater_unittest.cpp
index a2ba2df..85f86bc 100644
--- a/tools/test/tools_updater_unittest.cpp
+++ b/tools/test/tools_updater_unittest.cpp
@@ -110,7 +110,7 @@
     EXPECT_CALL(handler, verifyFile(ipmi_flash::updateBlobId))
         .WillOnce(Return(true));
 
-    updaterMain(&handler, image, signature);
+    updaterMain(&handler, image, signature, "static");
 }
 
 TEST_F(UpdaterTest, UpdateMainCleansUpOnFailure)
@@ -126,7 +126,8 @@
         .WillOnce(Return(false));
     EXPECT_CALL(handler, cleanArtifacts()).WillOnce(Return());
 
-    EXPECT_THROW(updaterMain(&handler, image, signature), ToolException);
+    EXPECT_THROW(updaterMain(&handler, image, signature, "static"),
+                 ToolException);
 }
 
 } // namespace host_tool