improving test coverage