Correctly set the allocator