Correct AttributeError

--HG--
branch : distribute
Merge fix for AttributeError

--HG--
branch : distribute