WIP fix bug in RandomModule.