Add linearity restriction