Fixed up test case