fix minor bug in test case