diff --git a/mock_system_state.h b/mock_system_state.h
index eccfd9b..e03546e 100644
--- a/mock_system_state.h
+++ b/mock_system_state.h
@@ -30,6 +30,7 @@
   virtual ~MockSystemState();
 
   MOCK_METHOD0(IsOOBEComplete, bool());
+  MOCK_METHOD0(IsOfficialBuild, bool());
 
   MOCK_METHOD1(set_device_policy, void(const policy::DevicePolicy*));
   MOCK_CONST_METHOD0(device_policy, const policy::DevicePolicy*());