removed hardcoded 0 prediction