diff --git a/dnn/test/CMakeLists.txt b/dnn/test/CMakeLists.txt
index 6d42880d..07bd7803 100644
--- a/dnn/test/CMakeLists.txt
+++ b/dnn/test/CMakeLists.txt
@@ -11,7 +11,7 @@ if(NOT ${MGE_ARCH} STREQUAL "naive")
     list(APPEND SOURCES ${SOURCES_})
     if(${MGE_ARCH} STREQUAL "fallback")
         message(WARNING "build only with fallback")
-    elseif(${MGE_ARCH} STREQUAL "x86")
+    elseif(${MGE_ARCH} STREQUAL "x86_64" OR ${MGE_ARCH} STREQUAL "i386")
         file(GLOB_RECURSE SOURCES_ x86/*.cpp)
         list(APPEND SOURCES ${SOURCES_})
     endif()