Update TH code for the new Heist