fixed imports and deleted test