diff --git a/dags/test-train-pytorch.py b/dags/test-train-pytorch.py
index 40b99bb..720a87f 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-0168-worker-gpu-0.airflow-k8s-0168-worker-gpu"
+MASTER_ADDR = "airflow-worker-gpu-0.airflow-worker-gpu"
 MASTER_PORT = "29500"
 NCCL_TIMEOUT = 1800