Minor fix in layout component