fix broken test case