diff --git a/capsulflask/spoke_model.py b/capsulflask/spoke_model.py
index 1afa0a2..bbae69b 100644
--- a/capsulflask/spoke_model.py
+++ b/capsulflask/spoke_model.py
@@ -205,10 +205,10 @@ class ShellScriptSpoke(VirtualizationInterface):
 
     networks = dict()
     for vm in vms_by_id.values():
-      if vm['network'] not in networks:
-        networks[vm['network']] = []
+      if vm['network_name'] not in networks:
+        networks[vm['network_name']] = []
 
-      networks[vm['network']].append(vm)
+      networks[vm['network_name']].append(vm)
 
     to_return = dict()
     to_return[current_app.config['SPOKE_HOST_ID']] = networks