Small layouting bug fix