Fixed prodiction bug.