Updated to match changed api in unitrie.