Unable to change Actor Coloration View Modes in PIE

When in PIE changing Actor Coloration view modes from the viewport visualisation mode [Liit] -> [Actor Coloration] -> ...

Does not update correctly when switching view mode and sometimes isn't correct for the view mode selected.

From reviewing the code in EditorViewportClient.cpp : ~2925
void FEditorViewportClient::ChangeActorColorationVisualizationMode(FName InName)

It uses the GWorld pointer, SetActivePrimitiveColorHandler(), changing this to GetWorld() appears to fix the issue.

Create a basic level with some basic static geometry, configured in a way such that any two actor colorizers would show different colors ( I placed 2 basic shapes, one in a sub level and one that affects/doesn't affect nav mesh)

PIE

Eject [F8] to the Debug Camera

Select a Actor Coloration -> Levels visualizer, and notice the colors are correct

Swap to your second visualizer Actor Coloration -> Affects Nav mesh and notice that the visualisation doesn't change.