added test coverage