Since you said you were using Studio One, one thing that you can do after recording the MIDI from your drum module is to use "Explode Pitches to Tracks" in Studio One which will give you separate MIDI tracks for each piece of the kit.