Small code refactor.