Fix the build of test code