HW2 test case fixes