Little fix in layout