add missing import
Merge pull request #426 from guillermooo/fix-import

add missing import