Reflect composite models modes in query editor

With composite models enabled, it would be great to be able to switch between the "Import", "Dual" and "Direct Query" while editing the source from Power Query Editor.
At the moment, editing the source step only displays "Import" and "Direct Query".
Sometimes, editing the source query for a table in Direct Query mode, the editor will display import. After validating the query the mode may or may not effectively change the storage mode.
(I couldn't reproduce the issue anymore while posting the idea).