Improve junit code coverage