diff --git a/tests/unit/hsm_mock.h b/tests/unit/hsm_mock.h
index 7a87342..db4b3f1 100644
--- a/tests/unit/hsm_mock.h
+++ b/tests/unit/hsm_mock.h
@@ -54,12 +54,12 @@ class HSMMock final : public HSM
                  openssl::SSL_EVP_PKEY_Ptr(const RSASpec &spec,
                                            const std::string &keyLabel,
                                            const std::vector<uint8_t> &keyID,
-                                           const HsmKeyParams &params));
+                                           const HsmKeyParameters &params));
     MOCK_METHOD4(generateKey,
                  openssl::SSL_EVP_PKEY_Ptr(const ECCSpec &spec,
                                            const std::string &keyLabel,
                                            const std::vector<uint8_t> &keyID,
-                                           const HsmKeyParams &params));
+                                           const HsmKeyParameters &params));
 };
 
 }  // namespace mococrw