Fix: incorrect unit count