fix the typo fix in linearize.py