updated for new test