Simple code refactor.