diff --git a/mock_eap_credentials.h b/mock_eap_credentials.h
index 3d14d68..43991da 100644
--- a/mock_eap_credentials.h
+++ b/mock_eap_credentials.h
@@ -30,7 +30,7 @@
   MOCK_CONST_METHOD3(Save, void(
       StoreInterface *store, const std::string &id, bool save_credentials));
   MOCK_METHOD0(Reset, void());
-  MOCK_METHOD2(SetKeyManagement, void(const std::string &key_management,
+  MOCK_METHOD2(SetKeyManagement, bool(const std::string &key_management,
                                       Error *error));
   MOCK_CONST_METHOD0(identity, const std::string &());
   MOCK_CONST_METHOD0(key_management, const std::string &());