Fix the test file to avoid build error.