Fix compilation error in the tests