Fix Comment after refactoring