Correcting draft checks
Fixes regression