fixed minor bug in test case