update tests for changes