Fix one function reference
Improve line updating script