Quick correction for refactoring error.