Small fix to improve functionality of network