Fix failed test case