Finish hw7 small test fix