Add ability to inject memory helper