you can move the return in front of the switch