bug fix --KW