lab2: bug fix