Fixes #26
Merge pull request #27 from ben-ng/draft-fix

Fixes #26