Fix compilation error due to interface changes