Minor fix

Remove unused import statement