Fixed missing import
Merge pull request #1271 from matthewprenger/master

Fixed missing import