Increase code coverage

:construction: