fixed minor bug
bug fix