Fixing missing imports
Merge pull request #12 from edx/e0d/fix-imports

Fixing missing imports