diff --git a/rust/cubesql/cubesql/src/sql/dataframe.rs b/rust/cubesql/cubesql/src/sql/dataframe.rs
index 11e507eb2ce0c..3c0f856f13a2c 100644
--- a/rust/cubesql/cubesql/src/sql/dataframe.rs
+++ b/rust/cubesql/cubesql/src/sql/dataframe.rs
@@ -699,15 +699,12 @@ pub fn batches_to_dataframe(
 mod tests {
     use std::sync::Arc;
 
-    use datafusion::{
-        arrow::{array::PrimitiveArray, datatypes::TimestampMicrosecondType},
-        scalar::ScalarValue::TimestampMillisecond,
-    };
+    use datafusion::arrow::array::PrimitiveArray;
     use itertools::Itertools;
 
     use super::*;
     use crate::compile::arrow::{
-        datatypes::{ArrowPrimitiveType, Field, TimestampNanosecondType},
+        datatypes::{ArrowPrimitiveType, Field},
         record_batch::RecordBatchOptions,
     };