From fee271b05527a62ad84515f8a78b7e13b27c7a03 Mon Sep 17 00:00:00 2001
From: Bernard Cooke <bernard-cooke@hotmail.com>
Date: Mon, 5 Jun 2023 20:13:44 +0100
Subject: [PATCH] fix: correct WSTransport.receive output

---
 stomp/adapter/ws.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/stomp/adapter/ws.py b/stomp/adapter/ws.py
index c2ae4556..ab645c49 100644
--- a/stomp/adapter/ws.py
+++ b/stomp/adapter/ws.py
@@ -212,7 +212,8 @@ def receive(self):
         :rtype: bytes
         """
         try:
-            return self.socket.recv().encode()
+            ret = self.socket.recv()
+            return ret.encode() if type(ret) == str else ret
         except socket.error:
             _, e, _ = sys.exc_info()
             if get_errno(e) in (errno.EAGAIN, errno.EINTR):