Fix TET cases for toolkit