Correct bug in test case