Index: trunk/tests/test_loader.py
===================================================================
--- trunk/tests/test_loader.py	(revision 23)
+++ trunk/tests/test_loader.py	(revision 49)
@@ -16,13 +16,15 @@
 
 try:
-    import sets
-except ImportError:
-    class _sets:
-        def Set(self, items):
+    Set = set
+except:
+    try:
+        from sets import Set
+    except ImportError:
+        def Set(items):
             set = {}
             for items in items:
                 set[items] = None
             return set
-    sets = _sets()
+
 
 INF = 1e300000
@@ -128,5 +130,5 @@
 """, {
 #    'baseball players': sets.Set(['Mark McGwire', 'Sammy Sosa', 'Ken Griffey']),
-    'baseball teams': sets.Set(['Boston Red Sox', 'Detroit Tigers', 'New York Yankees']),
+    'baseball teams': Set(['Boston Red Sox', 'Detroit Tigers', 'New York Yankees']),
 }