fix bug in the layout