fixing predict bug still