update tests to pass