Cleans up unused imports in unit tests