Fixing small layout issue