fixed mini bugs