Fix a little bit of layout