diff --git a/gateway/src/node/client_handling/websocket/connection_handler.rs b/gateway/src/node/client_handling/websocket/connection_handler.rs index 2047a902ea..ad29df86a7 100644 --- a/gateway/src/node/client_handling/websocket/connection_handler.rs +++ b/gateway/src/node/client_handling/websocket/connection_handler.rs @@ -327,7 +327,7 @@ impl Handle { let (res_sender, res_receiver) = oneshot::channel(); let clients_handler_request = ClientsHandlerRequest::Register( remote_address.clone(), - derived_shared_key, + derived_shared_key.clone(), mix_sender, res_sender, ); @@ -341,6 +341,9 @@ impl Handle { // managed to complete registration handshake) ClientsHandlerResponse::Register(status) => { self.remote_address = Some(remote_address); + if status { + self.shared_key = Some(derived_shared_key); + } ServerResponse::Register { status } } ClientsHandlerResponse::Error(e) => {