small layout fix