Fix broken test case