diff --git a/src/database/connection.rs b/src/database/connection.rs
index d2283caf..2c65fa28 100644
--- a/src/database/connection.rs
+++ b/src/database/connection.rs
@@ -3,7 +3,10 @@ use crate::{
     StatementBuilder, TransactionError,
 };
 use sea_query::{MysqlQueryBuilder, PostgresQueryBuilder, QueryBuilder, SqliteQueryBuilder};
-use std::{future::Future, pin::Pin, sync::Arc};
+use std::{future::Future, pin::Pin};
+
+#[cfg(feature = "mock")]
+use std::sync::Arc;
 
 #[cfg_attr(not(feature = "mock"), derive(Clone))]
 pub enum DatabaseConnection {
diff --git a/src/database/stream/query.rs b/src/database/stream/query.rs
index 73668da0..8383659a 100644
--- a/src/database/stream/query.rs
+++ b/src/database/stream/query.rs
@@ -1,4 +1,7 @@
-use std::{pin::Pin, sync::Arc, task::Poll};
+use std::{pin::Pin, task::Poll};
+
+#[cfg(feature = "mock")]
+use std::sync::Arc;
 
 use futures::Stream;
 #[cfg(feature = "sqlx-dep")]