Update for latest pgx v2