Correct heating during end-cap shrinking