diff --git a/socket_detector.py b/socket_detector.py
index 3fc2685..0c3d490 100644
--- a/socket_detector.py
+++ b/socket_detector.py
@@ -55,11 +55,10 @@ def main():
             socket_send.send(b'G')
         elif (size_buff[4] == 2):
             str_size = size_buff[0] << 8 | size_buff[1]
-            select_model = socket_receive.recv(str_size)
+            select_model = socket_receive.recv(str_size).decode("ascii")
             model_path = os.path.join("models", select_model)
             detector = WoodClass(w=4096, h=1200, n=3000, debug_mode=False)
             detector.load(path=model_path)
-            break