add some unit test coverage