Adds missing import.
Merge pull request #1101 from gradha/pr_adds_missing_import

Adds missing import.