Fix bad test cases

I didn't understand the problem...