[TiTools2] Bug fixes