try to improve TM

small change