Cleaned up a layout issue