diff --git a/buffet/manager.h b/buffet/manager.h
index a00ba20..3cf7021 100644
--- a/buffet/manager.h
+++ b/buffet/manager.h
@@ -8,6 +8,7 @@
 #include <map>
 #include <memory>
 #include <string>
+#include <vector>
 
 #include <base/files/file_path.h>
 #include <base/macros.h>
@@ -72,6 +73,10 @@
                   const std::string& json_command) override;
   void GetCommand(DBusMethodResponse<std::string> response,
                   const std::string& id) override;
+  void SetCommandVisibility(
+      scoped_ptr<chromeos::dbus_utils::DBusMethodResponse<>> response,
+      const std::vector<std::string>& in_names,
+      const std::string& in_visibility) override;
   std::string TestMethod(const std::string& message) override;
 
   void OnCommandDefsChanged();