diff --git a/library/gcm.c b/library/gcm.c
index ac6b945..20d55c0 100644
--- a/library/gcm.c
+++ b/library/gcm.c
@@ -659,7 +659,7 @@
 
 #endif /* !MBEDTLS_GCM_ALT */
 
-#if defined(MBEDTLS_SELF_TEST) && defined(MBEDTLS_AES_C)
+#if defined(MBEDTLS_SELF_TEST) && defined(MBEDTLS_CCM_GCM_CAN_AES)
 /*
  * AES-GCM test vectors from:
  *