Fix: Update weights to infer the label more correctly than before.
Merge branch 'fix/update-weight'