Fixed the compilation problem in the test code.