Fix up broken test case