update tests for full coverage