handling error on prediction slightly better