Build fix for function signature changes