flex box bug fix