How to modify the Distribution type