fix TG tests