Fixed prediction issue