Fixed a refactoring error..