layout bug fix --KW