Removed unused imports in test_wcs