fixed another hw2 bug