Missed thing after refactoring