Fixed up broken test case