minor bug fix in test case