Fix TSRM usage