To the extent that any adjustment under part II involves a change under section
704 in a partner’s distributive share of the amount of any partnership item shown on the partnership return, such adjustment shall be taken into account in applying this title to such partner for the partner’s taxable year for which such item was required to be taken into account.