mom-59 Use correct axes in diagnostics