Added missing import in memory.hooks