diff --git a/torchx/examples/apps/compute_world_size/config/__init__.py b/torchx/examples/apps/compute_world_size/config/__init__.py
index e69de29bb..2e41cd717 100644
--- a/torchx/examples/apps/compute_world_size/config/__init__.py
+++ b/torchx/examples/apps/compute_world_size/config/__init__.py
@@ -0,0 +1,5 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
diff --git a/torchx/examples/apps/compute_world_size/module/test/util_test.py b/torchx/examples/apps/compute_world_size/module/test/util_test.py
index a13fd8c9d..e2d0388de 100644
--- a/torchx/examples/apps/compute_world_size/module/test/util_test.py
+++ b/torchx/examples/apps/compute_world_size/module/test/util_test.py
@@ -26,8 +26,8 @@ def test_compute_world_size(self) -> None:
                     "rank": 0,
                     "world_size": 1,
                     "master_addr": "localhost",
-                    # ephemeral port range in linux
-                    "master_port": random.randint(32768, 60999),
+                    # specifying 0 as master_port makes TCPStore chose a free random port
+                    "master_port": 0,
                     "backend": "gloo",
                 }
             }