Adds missing import
Merge pull request #703 from learning-unlimited/import-fix

Adds missing import