Improved the code coverage