Removed old imports in tests