Improved unit tests to get more code coverage