[Minor] Fixed unused imports