fix bug predicitionui