added minor fix on the layout