Improves test coverage

Imporves the test coverage to 100%.