Add some test coverage