diff --git a/src/executor/query.rs b/src/executor/query.rs
index a96292e7..ba8c8169 100644
--- a/src/executor/query.rs
+++ b/src/executor/query.rs
@@ -1,4 +1,6 @@
-use crate::{debug_print, DbErr};
+#[cfg(feature = "mock")]
+use crate::debug_print;
+use crate::DbErr;
 use std::fmt;
 
 #[derive(Debug)]