Global heat balance and heat uptake in potential temperature coordinates