Switch on suitable midi part when switching track versions

Currently when we work with midi/audio part in Key/audio editor and switching to another track version nothing happens. Both regular and lower zone editors disappear and we have to choose corresponding part again.
Also, we often use track versions to compare different variants of melodies and I’d like immediately to see what the difference is.
So it would be quite handy if while switching to another track version the part which located in same project position will appear in editor.
Hope it makes sense,