JS bug fix --KW