Remove unused import from test