How can I get key edit to play the instrument I am editing

Does anyone here with work with MIDI?
How do I fix it so that the notes that have been selected for editing play back the right instrument?
At the moment key edit only auditions the instrument that has been selected on the project page.