Fixed syntax error from refactoring