Update distribution.py

of course it's sometimes one type, sometimes another