Fixes to TR update