Fix bugs in heur cb