Fix incorrect size for allocation.