clean code for test coverage