Remove stop_machine during module load v2