small fix to accomodate change to api