fixed the units exception