5.7 to 6.1 oil pump upgrade

Is it just the chain I should upgrade or are the sprockets part of the upgrade as well?