diff --git a/dags/test-train-pytorch.py b/dags/test-train-pytorch.py
index 5c73d3e..9a5f8c7 100644
--- a/dags/test-train-pytorch.py
+++ b/dags/test-train-pytorch.py
@@ -7,7 +7,7 @@ import json
 
 # --- CONFIGURATION ---
 WORLD_SIZE = 2
-MASTER_ADDR = "airflow-k8s-demo-0164-worker-gpu-0.airflow-k8s-demo-0164-worker-gpu"
+MASTER_ADDR = "airflow-k8s-demo-0165-worker-gpu-0.airflow-k8s-demo-0165-worker-gpu"
 MASTER_PORT = "29500"
 NCCL_TIMEOUT = 1800