diff --git a/src/primaite/config/_package_data/data_manipulation_marl.yaml b/src/primaite/config/_package_data/data_manipulation_marl.yaml
index 653ddfd3..deb43eea 100644
--- a/src/primaite/config/_package_data/data_manipulation_marl.yaml
+++ b/src/primaite/config/_package_data/data_manipulation_marl.yaml
@@ -1476,7 +1476,7 @@ simulation:
         options:
           db_server_ip: 192.168.1.14
       services:
-      - ty DNSClient
+      - type: DNSClient
 
 
 
diff --git a/src/primaite/notebooks/Data-Manipulation-Customising-Red-Agent.ipynb b/src/primaite/notebooks/Data-Manipulation-Customising-Red-Agent.ipynb
index 56e9bf5a..74a1e0ef 100644
--- a/src/primaite/notebooks/Data-Manipulation-Customising-Red-Agent.ipynb
+++ b/src/primaite/notebooks/Data-Manipulation-Customising-Red-Agent.ipynb
@@ -362,7 +362,7 @@
     "    cfg = yaml.safe_load(f)\n",
     "    cfg['simulation']['network']\n",
     "    for node in cfg['simulation']['network']['nodes']:\n",
-    "      if node['ref'] in ['client_1', 'client_2']:\n",
+    "      if node['hostname'] in ['client_1', 'client_2']:\n",
     "        node['applications'] = change['applications']\n",
     "\n",
     "env = PrimaiteGymEnv(game_config = cfg)\n",
@@ -407,7 +407,7 @@
     "    cfg = yaml.safe_load(f)\n",
     "    cfg['simulation']['network']\n",
     "    for node in cfg['simulation']['network']['nodes']:\n",
-    "      if node['ref'] in ['client_1', 'client_2']:\n",
+    "      if node['hostname'] in ['client_1', 'client_2']:\n",
     "        node['applications'] = change['applications']\n",
     "\n",
     "env = PrimaiteGymEnv(game_config = cfg)\n",