1 files changed, 3 insertions, 3 deletions
diff --git a/src/transport/mod.rs b/src/transport/mod.rs
index cf49d31..b4c512b 100644
--- a/src/transport/mod.rs
+++ b/src/transport/mod.rs
@@ -1,4 +1,4 @@
-use crate::encoding::{request::Request, response::Response};
+use crate::encoding::Message;
 use anyhow::Result;
 
 pub mod tcp;
@@ -6,6 +6,6 @@ pub mod udp;
 
 #[allow(async_fn_in_trait)]
 pub trait Transport {
-    async fn recv(&self) -> Result<Response>;
-    async fn send(&self, request: Request) -> Result<()>;
+    async fn recv(&self) -> Result<Message>;
+    async fn send(&self, message: Message) -> Result<()>;
 }