Fix compilation error in test