increased unit test coverage