Yes, there is. In the Toolbar, you can select, which MIDI part should be currently active.
See attached screenshot, please.
Btw: I highly recommended to use colours of Part in the Key Editor, in this case (green on my screenshot). Then you have even colour-coding for your instruments.
