Update for new Entropy API