diff --git a/src/bench/sink_bench/main.rs b/src/bench/sink_bench/main.rs
index 2341c9f82a144..9c01be3325081 100644
--- a/src/bench/sink_bench/main.rs
+++ b/src/bench/sink_bench/main.rs
@@ -56,7 +56,7 @@ use risingwave_connector::sink::{
     SINK_TYPE_APPEND_ONLY, SINK_TYPE_UPSERT,
 };
 use risingwave_connector::source::datagen::{
-    DatagenProperties, DatagenSplit, DatagenSplitEnumerator, DatagenSplitReader,
+    DatagenProperties, DatagenSplitEnumerator, DatagenSplitReader,
 };
 use risingwave_connector::source::{
     Column, DataType, SourceContext, SourceEnumeratorContext, SplitEnumerator, SplitReader,
@@ -304,7 +304,7 @@ impl MockDatagenSource {
             datagen_split_readers.push(
                 DatagenSplitReader::new(
                     properties.clone(),
-                    vec![DatagenSplit::try_from(splits).unwrap()],
+                    vec![splits],
                     parser_config.clone(),
                     SourceContext::dummy().into(),
                     Some(source_schema.clone()),
diff --git a/src/connector/src/macros.rs b/src/connector/src/macros.rs
index d39ec665bf2da..74b075493eb98 100644
--- a/src/connector/src/macros.rs
+++ b/src/connector/src/macros.rs
@@ -320,7 +320,7 @@ macro_rules! impl_connector_properties {
         $crate::paste! {
             impl ConnectorProperties {
                 pub async fn create_split_enumerator(self, context: $crate::source::base::SourceEnumeratorContextRef) -> $crate::error::ConnectorResult<Box<dyn AnySplitEnumerator>> {
-                    use crate::source::prelude::*;
+                    use $crate::source::prelude::*;
                     let enumerator: Box<dyn AnySplitEnumerator> = match self {
                         $(
                             ConnectorProperties::$variant_name(prop) =>