Unit counts are now correct

(easy fix thanks to peterm)