diff --git a/programs/aes/crypt_and_hash.c b/programs/aes/crypt_and_hash.c
index f64c341..6f4bfa3 100644
--- a/programs/aes/crypt_and_hash.c
+++ b/programs/aes/crypt_and_hash.c
@@ -310,7 +310,7 @@
 
         memset( key, 0, sizeof( key ) );
 
-        if( mbedtls_cipher_setkey( &cipher_ctx, digest, cipher_info->key_length,
+        if( mbedtls_cipher_setkey( &cipher_ctx, digest, cipher_info->key_bitlen,
                            MBEDTLS_ENCRYPT ) != 0 )
         {
             mbedtls_fprintf( stderr, "mbedtls_cipher_setkey() returned error\n");
@@ -442,7 +442,7 @@
 
         memset( key, 0, sizeof( key ) );
 
-        if( mbedtls_cipher_setkey( &cipher_ctx, digest, cipher_info->key_length,
+        if( mbedtls_cipher_setkey( &cipher_ctx, digest, cipher_info->key_bitlen,
                            MBEDTLS_DECRYPT ) != 0 )
         {
             mbedtls_fprintf( stderr, "mbedtls_cipher_setkey() returned error\n" );