Add MT assign mass support