Quick fix for network mode

Fixed bug where network mode would terminate immediately without waiting
for nodes to finish (indefinitely).