diff --git a/payload_state_unittest.cc b/payload_state_unittest.cc
index 2be5896..e7b7ed2 100644
--- a/payload_state_unittest.cc
+++ b/payload_state_unittest.cc
@@ -668,6 +668,12 @@
   EXPECT_CALL(*mock_system_state.mock_metrics_lib(), SendToUMA(
       "Installer.UpdateURLSwitches",
       2, _, _, _));
+  EXPECT_CALL(*mock_system_state.mock_metrics_lib(), SendToUMA(
+      "Installer.UpdateDurationMinutes",
+      _, _, _, _));
+  EXPECT_CALL(*mock_system_state.mock_metrics_lib(), SendToUMA(
+      "Installer.UpdateDurationUptimeMinutes",
+      _, _, _, _));
 
   payload_state.UpdateSucceeded();