Fixed a major bug that was causing predictions to not be saved.