Fix first test case