Minor updates to allocator.