Fixed add unit errors