Bug fixing: prediction has more related info