Small fix for line counter