Apologies for the late reply! the energy colour issue fix seems to have worked!
Here's a video demonstration of it but other than that I'm afraid I don't have anymore steps for repro for this issue.