Small fixes to TSMatMul