Increase coverage

TODO: 97% => 100%