Missing line after refactor