Fixed some issues with sampling of MoGSM.