updates to make the module work with changes for API testing