Fix bug in TM