fixing prediction