[Minor] Fixed unused imports
Merge pull request #279 from SofHad/unused-imports

[Minor] Fixed unused imports