Clear up a missed variable in the refactor.