hw2 ws bug fix