Minor fix
Removing unused imports