fix small layout bug