Fix test error: global name 'UnclaimedDraft' is not defined