Add missing flex wrapper div