Fix test case bug