Fix __setstate__ method of UnitsContainer. (Tests pass locally)