Increase max heap for stage