Fix bugs with label propagation