fix mini bugs