fix false balance value when prediction is 0