Fixed an error in hw1