improve mock test coverage