Add missing import

JIRA: FOR-523
Merge pull request #2907 from edx/gprice/FOR-523

Add missing import