Improved the tests to increase the code coverage.