fixed linearization issue