In its final judgment or other order, the court shall assign items to the surviving spouse and distributee to conform with the exchange under s. 766.31 (3) (b) to the extent that the court approved the exchange.