Fix a small bug in draw_net.py.