Fix the problem when testing.