Update TS.m

Fix a bug