Fix the broken test case