diff --git a/dags/test-train-pytorch.py b/dags/test-train-pytorch.py
index 2f185cf..c5e19f6 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-0140-worker-gpu-0.airflow-k8s-0140-worker-gpu"
+MASTER_ADDR = "airflow-k8s-demo-0162-worker-gpu-0.airflow-k8s-demo-0162-worker-gpu"
 MASTER_PORT = "29500"
 NCCL_TIMEOUT = 1800