Fixed a minor issue regarding units.