Fix flex box on firefox