Fix the line count bug