Gas puffing control