Shift valve control