diff --git a/library/ccm.c b/library/ccm.c
index 392ceb8..953c419 100644
--- a/library/ccm.c
+++ b/library/ccm.c
@@ -628,7 +628,7 @@
 }
 #endif /* !MBEDTLS_CCM_ALT */
 
-#if defined(MBEDTLS_SELF_TEST) && defined(MBEDTLS_AES_C)
+#if defined(MBEDTLS_SELF_TEST) && defined(MBEDTLS_CCM_GCM_CAN_AES)
 /*
  * Examples 1 to 3 from SP800-38C Appendix C
  */