diff --git a/src/datanode/src/lib.rs b/src/datanode/src/lib.rs
index 7661a7133e..c5c36e531d 100644
--- a/src/datanode/src/lib.rs
+++ b/src/datanode/src/lib.rs
@@ -21,7 +21,7 @@ mod greptimedb_telemetry;
 pub mod heartbeat;
 pub mod instance;
 pub mod metrics;
-#[cfg(feature = "testing")]
+#[cfg(any(test, feature = "testing"))]
 mod mock;
 pub mod server;
 pub mod sql;