Improve coding style re #9556

Signed-off-by: Karl Palmen <karl.palmen@stfc.ac.uk>
Merge branch 'feature/9556_muon_dkt' into develop