fix the test fail