A little more unit testing coverage