Remove uneeded import
Reversion refactoring