Change the allocation size calculation

This is a bit more opaque.