diff --git a/dnn/src/common/batch_normalization.cpp b/dnn/src/common/batch_normalization.cpp
index a79c0f39839fb0a93a63ac310100cc8f5a826ede..b1f049f7630f83d00d3333c61b3ff2d127e2fe6d 100644
--- a/dnn/src/common/batch_normalization.cpp
+++ b/dnn/src/common/batch_normalization.cpp
@@ -58,6 +58,7 @@ void BNBackward::check_exec(const TensorLayout& x, const TensorLayout& dy,
             get_workspace_in_bytes(x, dy, saved_batch_mean, saved_batch_variance,
                                    bn_scale, d_bn_scale, d_bn_bias, dx);
     megdnn_assert(workspace_in_bytes >= required_workspace_in_bytes);
+    megdnn_assert(param().fwd_mode == Param::FwdMode::TRAINING, "BNBackward only support TRAINING mode");
 }
 
 }  // namespace megdnn
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 40ed0dc3bc2855492dd521b3d5ded8b9764d1306..d4bedb37066f23b6f0ce82608fc6cd0ac3958283 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -156,7 +156,7 @@ if(ANDROID)
     target_link_libraries(megbrain PUBLIC log)
 endif()
 
-if(NOT MGE_BUILD_XXX)
+if(NOT MGE_BUILD_IMPERATIVE_RT)
     # Build as SHARED or STATIC depending on BUILD_SHARED_LIBS=ON/OFF
     add_library(megengine)
     target_link_libraries(megengine PUBLIC megbrain megdnn)