diff --git a/socket_detector.py b/socket_detector.py
index c2a7852..d28f246 100644
--- a/socket_detector.py
+++ b/socket_detector.py
@@ -34,7 +34,7 @@ def process_cmd(cmd: str, data: any, connected_sock: socket.socket, detector: Wo
     elif cmd == 'TR':
         detector = WoodClass(w=4096, h=1200, n=3000, debug_mode=False)
         model_name = None
-        if "/n" in data:
+        if ":" in data:
             data, model_name = data.split(":", 1)
             model_name = model_name + ".p"
         settings.data_path = data