Draft mode fixed

Closes #34.
Merge pull request #35 from urdh/fix/draft

Draft mode fixed