Fixed memoize function