update tests to ensure passes