fixed minor bug fix