fix a test case, needs more tweak