Correcting bug with units