From dc7f788d493e75f4949e1f9a705a4479be537893 Mon Sep 17 00:00:00 2001
From: Noel Kwan <noelkwan1998@gmail.com>
Date: Mon, 25 Nov 2024 22:46:30 +0800
Subject: [PATCH] more madsim fix

---
 src/connector/src/connector_common/postgres.rs    | 1 +
 src/connector/src/source/cdc/external/postgres.rs | 1 -
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/connector/src/connector_common/postgres.rs b/src/connector/src/connector_common/postgres.rs
index 2701ec142b929..73c8241aae83e 100644
--- a/src/connector/src/connector_common/postgres.rs
+++ b/src/connector/src/connector_common/postgres.rs
@@ -30,6 +30,7 @@ use thiserror_ext::AsReport;
 use tokio_postgres::types::Kind as PgKind;
 use tokio_postgres::{Client as PgClient, NoTls};
 
+#[cfg(not(madsim))]
 use super::maybe_tls_connector::MaybeMakeTlsConnector;
 use crate::error::ConnectorResult;
 
diff --git a/src/connector/src/source/cdc/external/postgres.rs b/src/connector/src/source/cdc/external/postgres.rs
index d3684030030bf..ecc503934bb69 100644
--- a/src/connector/src/source/cdc/external/postgres.rs
+++ b/src/connector/src/source/cdc/external/postgres.rs
@@ -26,7 +26,6 @@ use serde_derive::{Deserialize, Serialize};
 use tokio_postgres::types::PgLsn;
 
 use crate::connector_common::create_pg_client;
-#[cfg(not(madsim))]
 use crate::error::{ConnectorError, ConnectorResult};
 use crate::parser::postgres_row_to_owned_row;
 use crate::parser::scalar_adapter::ScalarAdapter;