diff --git a/CMakeLists.txt b/CMakeLists.txt
index 6dff2631..6b4f22cd 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -80,6 +80,7 @@ endif()
 
 option(ENABLE_SM3_AVX_BMI2 "Enable SM3 AVX+BMI2 assembly implementation" OFF)
 if (ENABLE_SM3_AVX_BMI2)
+	add_definitions(-DSM3_AVX_BMI2)
 	enable_language(ASM)
 	list(APPEND src src/sm3_avx_bmi2.s)
 endif()
diff --git a/src/sm3.c b/src/sm3.c
index e68be8e8..196a88f8 100644
--- a/src/sm3.c
+++ b/src/sm3.c
@@ -152,7 +152,7 @@ static uint32_t K[64] = {
 	*/
 };
 
-#ifndef ENABLE_SM3_AVX_BMI2
+#ifndef SM3_AVX_BMI2
 void sm3_compress_blocks(uint32_t digest[8], const uint8_t *data, size_t blocks)
 {
 	uint32_t A;