minor bug fix, comments