minor fix to the layout