Fix compilation error
Fix compilation error.