- still supporting py2.6 here
diff --git a/test/ext/test_babelplugin.py b/test/ext/test_babelplugin.py
index abce70a..3658c50 100644
--- a/test/ext/test_babelplugin.py
+++ b/test/ext/test_babelplugin.py
@@ -82,12 +82,12 @@
     @skip()
     def test_extract_utf8(self):
         mako_tmpl = open(os.path.join(template_base, 'gettext_utf8.mako'), 'rb')
-        message = next(extract(mako_tmpl, {'_', None}, [], {'encoding': 'utf-8'}))
+        message = next(extract(mako_tmpl, set(['_', None]), [], {'encoding': 'utf-8'}))
         assert message == (1, '_', u'K\xf6ln', [])
 
     @skip()
     def test_extract_cp1251(self):
         mako_tmpl = open(os.path.join(template_base, 'gettext_cp1251.mako'), 'rb')
-        message = next(extract(mako_tmpl, {'_', None}, [], {'encoding': 'cp1251'}))
+        message = next(extract(mako_tmpl, set(['_', None]), [], {'encoding': 'cp1251'}))
         # "test" in Rusian. File encoding is cp1251 (aka "windows-1251")
         assert message == (1, '_', u'\u0442\u0435\u0441\u0442', [])