Fixed an assertion error in AbstractPredictionModel#onUpdate