Fix set(): it needs to skip @contextmanager.

diff --git a/__init__.py b/__init__.py
index bc205292e79719b98b91b5460204f423cb30602a..f4e6b9fe4773ffd780b564e20314a0cf222217c5 100644 (file)
--- a/__init__.py
+++ b/__init__.py
@@ -153,7 +153,7 @@ class connect(object):
 
     @contextmanager
     def set(self, statement):
-        with self.execute(statement, 1) as cursor:
+        with self.execute(statement, 2) as cursor:
             yield cursor
 
     def all(self, statement, context=None):