minor bug fix