fixed another incorrect assumption with screening in the non-linK implementation