fix the layout issue