Fix compilation errors with latest WAI/warp