Fix for Unit type