1 files changed, 4 insertions, 4 deletions
diff --git a/src/transport/udp.rs b/src/transport/udp.rs
index c86ce50..391e2b3 100644
--- a/src/transport/udp.rs
+++ b/src/transport/udp.rs
@@ -1,5 +1,5 @@
 use super::Transport;
-use crate::encoding::{request::Request, response::Response};
+use crate::encoding::Message;
 use anyhow::Result;
 use log::debug;
 use std::str::FromStr;
@@ -15,14 +15,14 @@ impl UdpTransport {
     }
 }
 impl Transport for UdpTransport {
-    async fn recv(&self) -> Result<Response> {
+    async fn recv(&self) -> Result<Message> {
         let mut buf = [0; 1024];
         let size = self.sock.recv(&mut buf).await?;
         let message = String::from_utf8(buf[..size].to_vec())?;
         debug!("{message}");
-        Response::from_str(message.trim_end())
+        Message::from_str(message.trim_end())
     }
-    async fn send(&self, request: Request) -> Result<()> {
+    async fn send(&self, request: Message) -> Result<()> {
         debug!("-> {request}");
         self.sock.send(format!("{request}").as_bytes()).await?;
         Ok(())