display flex layout correction