Add missing import
Merge pull request #3798 from dimagi/fix-staging

Add missing import