diff --git a/library/aes.c b/library/aes.c
index 03dabd8..e7a888f 100644
--- a/library/aes.c
+++ b/library/aes.c
@@ -675,6 +675,7 @@
 }
 #endif /* MBEDTLS_CIPHER_MODE_XTS */
 
+#if defined(MBEDTLS_AES_SCA_COUNTERMEASURES)
 static void mbedtls_generate_fake_key( unsigned int keybits, mbedtls_aes_context *ctx )
 {
     unsigned int qword;
@@ -684,6 +685,7 @@
         ctx->frk[ qword - 1 ] = mbedtls_platform_random_uint32();
     }
 }
+#endif /* MBEDTLS_AES_SCA_COUNTERMEASURES */
 
 /*
  * AES key schedule (encryption)
@@ -729,7 +731,9 @@
     else
 #endif
     ctx->rk = RK = ctx->buf;
+#if defined(MBEDTLS_AES_SCA_COUNTERMEASURES)
     mbedtls_generate_fake_key( keybits, ctx );
+#endif
 
 #if defined(MBEDTLS_AESNI_C) && defined(MBEDTLS_HAVE_X86_64)
     if( mbedtls_aesni_has_support( MBEDTLS_AESNI_AES ) )
@@ -869,7 +873,9 @@
     else
 #endif
     ctx->rk = RK = ctx->buf;
+#if defined(MBEDTLS_AES_SCA_COUNTERMEASURES)
     mbedtls_generate_fake_key( keybits, ctx );
+#endif
 
     /* Also checks keybits */
     if( ( ret = mbedtls_aes_setkey_enc( &cty, key, keybits ) ) != 0 )
@@ -1084,6 +1090,7 @@
 
     aes_data_real.rk_ptr = ctx->rk;
     aes_data_fake.rk_ptr = ctx->frk;
+
     aes_data_table[0] = &aes_data_real;
     aes_data_table[1] = &aes_data_fake;
 
@@ -1364,6 +1371,7 @@
 
     aes_data_real.rk_ptr = ctx->rk;
     aes_data_fake.rk_ptr = ctx->frk;
+
     aes_data_table[0] = &aes_data_real;
     aes_data_table[1] = &aes_data_fake;