diff --git a/buffet/dbus_command_proxy_unittest.cc b/buffet/dbus_command_proxy_unittest.cc
index 0aa701d..5225f11 100644
--- a/buffet/dbus_command_proxy_unittest.cc
+++ b/buffet/dbus_command_proxy_unittest.cc
@@ -79,6 +79,8 @@
     // Use WillRepeatedly because GetName is used for logging.
     EXPECT_CALL(*command_, GetName())
         .WillRepeatedly(ReturnRefOfCopy<std::string>("robot.jump"));
+    EXPECT_CALL(*command_, GetComponent())
+        .WillRepeatedly(ReturnRefOfCopy<std::string>("myComponent"));
     EXPECT_CALL(*command_, GetState())
         .WillRepeatedly(Return(weave::Command::State::kQueued));
     EXPECT_CALL(*command_, GetOrigin())
@@ -149,6 +151,7 @@
   EXPECT_EQ(VariantDictionary{}, GetCommandAdaptor()->GetProgress());
   EXPECT_EQ(VariantDictionary{}, GetCommandAdaptor()->GetResults());
   EXPECT_EQ("robot.jump", GetCommandAdaptor()->GetName());
+  EXPECT_EQ("myComponent", GetCommandAdaptor()->GetComponent());
   EXPECT_EQ(kTestCommandId, GetCommandAdaptor()->GetId());
 }