get_predictions fixed