diff --git a/include/psa/crypto_struct.h b/include/psa/crypto_struct.h
index 273f6b6..f89073b 100644
--- a/include/psa/crypto_struct.h
+++ b/include/psa/crypto_struct.h
@@ -333,6 +333,12 @@
     return( attributes->type );
 }
 
+static inline void psa_set_key_bits(psa_key_attributes_t *attributes,
+                                    size_t bits)
+{
+    attributes->bits = bits;
+}
+
 static inline size_t psa_get_key_bits(
     const psa_key_attributes_t *attributes)
 {