fix some codes in the test