fix missing import
Merge pull request #3233 from dimagi/import-error

fix missing import