fix pred bug