correct layout bug