Частенько вертска едет из-за отсутствия flex-bazis - вы можете задавать его в тех же величинах (в процентах лучше, конечно) и поддержать через width или max-width. Но с max-width - все темно - часто мобилы с сафари это свойство игнорируют как факт.
Еще хак - поиграйтесь с flex: 1 0 initial; - тут самое интересное именно initial.
