Fixed flexbox code.