fix samll bugs