fixed weird refactoring error?