update tests to cover new behavior