Fixed minor refactoring mistake.