As was mentioned in another answer, it was fixed in Android Studio 2.2 (bug report).
However, it still wasn’t working for me even in Android Studio 3.0
It started working only after I checked “Show alternative source switcher” in Settings.
screenshot