prediction problem fix