add test case
correct bug