Slight fix in linearizing