update tests to increase coverage