Fixing missed refactor.