Fix TR unit tests

Was broken by d4963473adecf771a16f49120f00fd9296acc6ff