Fix dismissing keyboard on transition