diff --git a/containers/cluster_node/cluster_worker.py b/containers/cluster_node/cluster_worker.py
index d4520fa..9e0fd25 100644
--- a/containers/cluster_node/cluster_worker.py
+++ b/containers/cluster_node/cluster_worker.py
@@ -105,6 +105,8 @@ class ClusterWorker:
 
         # 2. Write new files
         for filename, content in files.items():
+            if filename == 'wg-firewall.sh' and isinstance(content, str):
+                content = content.replace('wireguard-webadmin-dns', 'cluster-node-dns')
             filepath = os.path.join(WIREGUARD_DIR, filename)
             with open(filepath, 'w') as f:
                 f.write(content)