Minor tweak to allocator selection