Small fix for api compat