Fixed bug where getPredictionForInstance was not getting called.