Update for new version of Testium