TMS example code fix