Update grant_type_code Test

`grant_type_code` test fails when callbacks are involved (I tested with Postgres and MongoDB). Fixing that.