TS + bug fix