Added vendor format key to aes context
diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 53252ca..6affd85 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -2428,7 +2428,6 @@
     return( PSA_SUCCESS );
 }
 
-
 /****************************************************************/
 /* MAC */
 /****************************************************************/
@@ -2489,6 +2488,7 @@
     switch( key_type )
     {
         case PSA_KEY_TYPE_AES:
+        case PSA_KEY_TYPE_VENDOR_AES:
             cipher_id_tmp = MBEDTLS_CIPHER_ID_AES;
             break;
         case PSA_KEY_TYPE_DES:
@@ -3789,6 +3789,7 @@
         operation->iv_size = 12;
 #endif
 
+
 exit:
     if( status == 0 )
         status = mbedtls_to_psa_error( ret );