Update for alloc_buffer changes.