Minor change in layout to be consistent