Fixed the test bug