Fix: pass rng to distribution's __init__.