Fix networkx 1.9 compatibility issue