diff --git a/include/lib/psa/dice_protection_environment.h b/include/lib/psa/dice_protection_environment.h
index ddf9aa7..61b6482 100644
--- a/include/lib/psa/dice_protection_environment.h
+++ b/include/lib/psa/dice_protection_environment.h
@@ -41,6 +41,8 @@
  *
  * \param[in]  context_handle              Input context handle for the DPE
  *                                         context.
+ * \param[in]  cert_id                     Logical certificate id to which derived
+ *                                         context belongs to.
  * \param[in]  retain_parent_context       Flag to indicate whether to retain the
  *                                         parent context. True only if a client
  *                                         will call further DPE commands on the
@@ -77,6 +79,7 @@
  * \return Returns error code of type dpe_error_t
  */
 dpe_error_t dpe_derive_context(int      context_handle,
+			       uint32_t cert_id,
 			       bool     retain_parent_context,
 			       bool     allow_new_context_to_derive,
 			       bool     create_certificate,