Fix the predictions setup, remove index