added some more test coverage