Fixed --on --off handling