layout bug fix