Add missing import
fix pylint error
Merge pull request #8544 from UtahDave/develop

Fix pylint errors