Make shifts restore selection in visual state