Remove unused import in test
draftupdate