diff --git a/components/service/crypto/client/caller/packed-c/crypto_caller_key_derivation.h b/components/service/crypto/client/caller/packed-c/crypto_caller_key_derivation.h
index 496f07e..763e2af 100644
--- a/components/service/crypto/client/caller/packed-c/crypto_caller_key_derivation.h
+++ b/components/service/crypto/client/caller/packed-c/crypto_caller_key_derivation.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021, Arm Limited and Contributors. All rights reserved.
+ * Copyright (c) 2021-2022, Arm Limited and Contributors. All rights reserved.
  *
  * SPDX-License-Identifier: BSD-3-Clause
  */
@@ -524,7 +524,7 @@
 					}
 					else {
 						/* Insufficient buffer space */
-						psa_status = PSA_ERROR_INVALID_ARGUMENT;
+						psa_status = PSA_ERROR_BUFFER_TOO_SMALL;
 					}
 				}
 				else {