Important fix in the semi-supervised prediction with a huge impact in some experiments :)