fix the test bug