Small fix for the alloc buffers