Removed wrong import from test