fix a bit of layout