Small bug fix for pyretic_switch.py code