diff --git a/CMakeLists.txt b/CMakeLists.txt
index 97af6192cb899590927b80de9bda217c5ac56221..7a7b5860a122a853fc9ce1da6494fc039b38bc10 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -309,5 +309,5 @@ if (ON_INFER)
     add_definitions(-DPADDLE_ON_INFERENCE)
 else()
     #TODO(luotao), combine this warning with `make inference_lib_dist` command.
-    message(WARNING "On inference mode, will take place some specific optimization. Only used in make inference_lib_dist.")
+    message(WARNING "On inference mode, will take place some specific optimization. Turn on the ON_INFER flag when building inference_lib only.")
 endif()