Small fix in test case