diff --git a/src/protocol/tls.rs b/src/protocol/tls.rs
index c02c8b4..091092a 100644
--- a/src/protocol/tls.rs
+++ b/src/protocol/tls.rs
@@ -7,6 +7,7 @@
 #![allow(dead_code)]
 
 use crate::crypto::{sha256_hmac, SecureRandom};
+#[cfg(test)]
 use crate::error::ProxyError;
 use super::constants::*;
 use std::time::{SystemTime, UNIX_EPOCH};