fix small layout changes