Fixing refactoring error, oops!