Finish hw7 small test fix

hw7 pull request