Fix memoization test not being run