fixing callback args to conform to latest api