Minor fix to layout