Add more memory to distribution sanity test