Mini code refactor.