minor bug fix --KW