Fix arg type for mapping