Add some more test coverage