Fixed a reference error from my refactoring.