add missing import
Merge pull request #1484 from mrocklin/fix-import

add missing import