update tests to run with latest changes