1 files changed, 3 insertions, 1 deletions
diff --git a/src/components/functional_module/test/include/mock_application.h b/src/components/functional_module/test/include/mock_application.h
index e73851094e..95bd4a0505 100644
--- a/src/components/functional_module/test/include/mock_application.h
+++ b/src/components/functional_module/test/include/mock_application.h
@@ -115,6 +115,7 @@ class MockApplication : public Application {
                bool(mobile_apis::FunctionID::eType cmd_id,
                     TLimitSource source));
   MOCK_METHOD0(usage_report, UsageStatistics&());
+  MOCK_METHOD1(SetInitialState, void(HmiStatePtr state));
   MOCK_METHOD2(SubscribeToSoftButtons,
                void(int32_t cmd_id, const SoftButtonID& softbuttons_id));
   MOCK_METHOD1(IsSubscribedToSoftButton, bool(const uint32_t softbutton_id));
@@ -132,7 +133,8 @@ class MockApplication : public Application {
   MOCK_CONST_METHOD0(tbt_show_command, const smart_objects::SmartObject*());
   MOCK_CONST_METHOD0(SubscribedButtons,
                      const std::set<mobile_apis::ButtonName::eType>&());
-  MOCK_CONST_METHOD0(SubscribesIVI, const std::set<uint32_t>&());
+  MOCK_CONST_METHOD0(SubscribesIVI,
+                     const application_manager::VehicleInfoSubscriptions&());
   MOCK_CONST_METHOD0(keyboard_props, const smart_objects::SmartObject*());
   MOCK_CONST_METHOD0(menu_title, const smart_objects::SmartObject*());
   MOCK_CONST_METHOD0(menu_icon, const smart_objects::SmartObject*());