diff --git a/src/judas_server/backend/backend_server.py b/src/judas_server/backend/backend_server.py index b1bea71..fac89b1 100644 --- a/src/judas_server/backend/backend_server.py +++ b/src/judas_server/backend/backend_server.py @@ -106,7 +106,7 @@ class BackendServer: client (Client): The client to send the ACK to. target_id (str): The id of the ACK'd message. """ - ack: bytes = Message.ack(target_id=target_id).to_bytes() + ack: bytes = Message.Control.ack(target_id=target_id).to_bytes() self.logger.info(f"[>] Sending ACK to {client}") client.outbound += ack @@ -116,7 +116,7 @@ class BackendServer: Args: client (Client): The client to send the CLOSE message to. """ - close_msg: bytes = Message.close().to_bytes() + close_msg: bytes = Message.Control.close().to_bytes() self.logger.info(f"[>] Sending CLOSE to {client}") client.outbound += close_msg