Fix missed refactor.