fix assert in nlp linear