fixed the test case broke by previous commit