Minor fix to how much memory is allocated