Fix missing drain
Merge pull request #101 from mafintosh/patch-1

Fix missing drain when return value of write is undefined