small modify
bug fix