adjust for new test api