diff --git a/engines/python/setup/djl_python_engine.py b/engines/python/setup/djl_python_engine.py
index 94ee43093e..596ac43578 100644
--- a/engines/python/setup/djl_python_engine.py
+++ b/engines/python/setup/djl_python_engine.py
@@ -46,7 +46,6 @@ def __init__(self, args, service):
             os.environ["RANK"] = rank
 
         self.sock_type = args.sock_type
-        self.sock_name = f"{args.sock_name}.{rank}" if rank else args.sock_name
         self.port = args.port
         self.service = service
         self.device_id = args.device_id
@@ -56,14 +55,13 @@ def __init__(self, args, service):
         if self.sock_type == "unix":
             if self.sock_name is None:
                 raise ValueError("Missing sock-name argument.")
+            self.sock_name = f"{args.sock_name}.{rank}" if rank else args.sock_name
 
             self.clean_up()
         elif self.sock_type == "tcp":
-            if self.sock_name is None:
-                raise ValueError("Missing sock-name argument.")
             if self.port is None:
                 raise ValueError("Missing port argument.")
-            self.port = int(self.port) + int(rank)
+            self.port = int(self.port) + int(rank) if rank else self.port
         else:
             raise ValueError(f"Invalid socket-type: {self.sock_type}.")