fixed samll bug