samll bug fix