update references to treatment objects