Improve the drip code slightly