Fix the test bug