diff --git a/mantis/src/detection/ml/feature_extractor.rs b/mantis/src/detection/ml/feature_extractor.rs
index 32858e2..b56aae9 100644
--- a/mantis/src/detection/ml/feature_extractor.rs
+++ b/mantis/src/detection/ml/feature_extractor.rs
@@ -9,10 +9,6 @@ use crate::model::ml_detection::{ClipParams, PacketData};
 pub struct FlowFeatures {
     pub features: Vec<f64>,
     pub feature_num: usize,
-
-    pub src_ip: String,
-    pub dst_ip: String,
-    pub timestamp: u64,
 }
 
 impl FlowFeatures {
@@ -23,13 +19,7 @@ impl FlowFeatures {
         for name in feature_names {
             features.push(Self::get_feature_by_name(flow, name.trim()));
         }
-        Self {
-            features,
-            feature_num,
-            src_ip: flow.flow_key.src_ip.to_string(),
-            dst_ip: flow.flow_key.dst_ip.to_string(),
-            timestamp: flow.start_time_us,
-        }
+        Self { features, feature_num }
     }
 
     fn get_feature_by_name(flow: &FlowData, feature_name: &str) -> f64 {