Minor update for heap allocation