Fix error in layout