Update to work with latest axiom-memory-adapter