Flexbox fixes