Updating tests to increase code coverage