diff --git a/src/infrastructure/litellm.rs b/src/infrastructure/litellm.rs
index af367b1..622ac20 100644
--- a/src/infrastructure/litellm.rs
+++ b/src/infrastructure/litellm.rs
@@ -292,7 +292,7 @@ pub async fn get_litellm_usage(
     })
 }
 
-#[cfg(test)]
+#[cfg(all(test, feature = "server"))]
 mod tests {
     use super::*;