Add more local imports
Merge pull request #2007 from 9rnsr/fix_imports

Add more local imports