add missing import
Merge pull request #1606 from nden/add-missing-import

add missing import