fixing predict bugs
fixing predict bugs