diff --git a/buffet/states/state_manager.h b/buffet/states/state_manager.h
index cf2c60a..a34f7c1 100644
--- a/buffet/states/state_manager.h
+++ b/buffet/states/state_manager.h
@@ -61,6 +61,7 @@
   std::vector<StateChange> GetAndClearRecordedStateChanges();
 
  private:
+  friend class BaseApiHandlerTest;
   friend class StateManagerTest;
 
   // Updates a single property value. |full_property_name| must be the full