fixed bug in  and  and added handling of non-vector predictions