Fix flexbox issues with firefox