Fixed a little issue with layout