diff --git a/library/aes.c b/library/aes.c
index 9869be6aae..cf63303df0 100644
--- a/library/aes.c
+++ b/library/aes.c
@@ -322,7 +322,7 @@ static const uint32_t RT2[256] = { RT };
 static const uint32_t RT3[256] = { RT };
 #undef V
 
-#endif /* !defined(MBEDTLS_AES_DECRYPT_ALT) || !defined(MBEDTLS_AES_SETKEY_DEC) */
+#endif /* !defined(MBEDTLS_AES_DECRYPT_ALT) || !defined(MBEDTLS_AES_SETKEY_DEC_ALT) */
 
 #endif /* !MBEDTLS_AES_FEWER_TABLES */