diff --git a/src/query/src/part_sort.rs b/src/query/src/part_sort.rs
index 03005edb13..729bb6e3bb 100644
--- a/src/query/src/part_sort.rs
+++ b/src/query/src/part_sort.rs
@@ -341,6 +341,17 @@ impl PartSortStream {
             )?,
         );
 
+        match sort_column.data_type() {
+            arrow_schema::DataType::Timestamp(unit, _) => {
+                assert_eq!(cur_range.start.unit().as_arrow_time_unit(), *unit);
+                assert_eq!(cur_range.end.unit().as_arrow_time_unit(), *unit);
+            }
+            _ => panic!(
+                "Unsupported data type for sort column: {:?}",
+                sort_column.data_type()
+            ),
+        }
+
         for (idx, val) in sort_column_iter {
             // ignore vacant time index data
             if let Some(val) = val {