Supply the lacked local import
Merge pull request #2064 from 9rnsr/fix_imports

Supply the lacked local import