Fix broken TR test

Sorry, forgot to merge this into commit
892887b before pushing the last set of changes.