Update a test case
Fix my mistakes