diff --git a/xdl/xdl/data_io/data_io.cc b/xdl/xdl/data_io/data_io.cc
index adc0904c..0f573b7c 100644
--- a/xdl/xdl/data_io/data_io.cc
+++ b/xdl/xdl/data_io/data_io.cc
@@ -401,7 +401,7 @@ bool DataIO::GetUniqueIds() const {
 }
 
 void DataIO::SetSgroupQueueCapacity(size_t capacity) {
-  CHECK(!running_);
+  XDL_CHECK(!running_);
   sgroup_queue_capacity_ = capacity;
 }