Fix two test cases