added some test coverage