fixed the layout bug