fix samll bug