fix the bug in test