Small fix in network code