Fix problem with test case