Fix bugs to for dummy prediction