fix some bug of the test