diff --git a/src/promql/src/extension_plan/instant_manipulate.rs b/src/promql/src/extension_plan/instant_manipulate.rs
index 67f05961ff..65fe94bc9e 100644
--- a/src/promql/src/extension_plan/instant_manipulate.rs
+++ b/src/promql/src/extension_plan/instant_manipulate.rs
@@ -857,7 +857,7 @@ mod test {
         )
         .to_execution_plan(exec_input);
 
-        assert!(format!("{exec:?}").contains("reuse_all_non_sample_columns: true"));
+        assert!(format!("{exec:?}").contains("reuse_tsid_column: true"));
     }
 
     #[tokio::test]
diff --git a/src/query/src/promql/planner.rs b/src/query/src/promql/planner.rs
index 477bd7f5b3..6b17b7837a 100644
--- a/src/query/src/promql/planner.rs
+++ b/src/query/src/promql/planner.rs
@@ -4851,7 +4851,7 @@ mod test {
         let exec = manipulate.to_execution_plan(Arc::new(DataSourceExec::new(Arc::new(
             MemorySourceConfig::try_new(&[], Arc::new(ArrowSchema::empty()), None).unwrap(),
         ))));
-        assert!(format!("{exec:?}").contains("reuse_all_non_sample_columns: true"));
+        assert!(format!("{exec:?}").contains("reuse_tsid_column: true"));
     }
 
     #[tokio::test]