diff --git a/api/libs/agents/src/tools/categories/file_tools/modify_metrics.rs b/api/libs/agents/src/tools/categories/file_tools/modify_metrics.rs
index 83d538900..707e2e7a9 100644
--- a/api/libs/agents/src/tools/categories/file_tools/modify_metrics.rs
+++ b/api/libs/agents/src/tools/categories/file_tools/modify_metrics.rs
@@ -384,6 +384,8 @@ async fn get_metric_id_description() -> String {
 mod tests {
     use std::collections::HashMap;
 
+    use crate::tools::file_tools::common::{apply_modifications_to_content, Modification};
+
     use super::*;
     use chrono::Utc;
     use serde_json::json;