fix test failure. (for catching up the original code)
refine test code.