fixing a unit error