A little more unit test coverage