Another small layout fix