fixed a problem with the layout