Draft: small fix in Draft.py