add missing import
Merge pull request #5444 from ticoann/missing_import

add missing import