bugfix: prediction extraction