Update for new relations in TS.