diff --git a/library/ecp_invasive.h b/library/ecp_invasive.h
index 284626b689..050c218203 100644
--- a/library/ecp_invasive.h
+++ b/library/ecp_invasive.h
@@ -303,6 +303,6 @@ int mbedtls_ecp_modulus_setup(mbedtls_mpi_mod_modulus *N,
                               const mbedtls_ecp_modulus_type ctype);
 #endif /* MBEDTLS_ECP_WITH_MPI_UINT */
 
-#endif /* MBEDTLS_TEST_HOOKS && MBEDTLS_ECP_C */
+#endif /* MBEDTLS_TEST_HOOKS && MBEDTLS_ECP_LIGHT */
 
 #endif /* MBEDTLS_ECP_INVASIVE_H */