diff --git a/lib/async.js b/lib/async.js
index d2afd47b..25b02e64 100644
--- a/lib/async.js
+++ b/lib/async.js
@@ -20,11 +20,7 @@ function asyncAdapter(generator) {
                 });
         }
 
-        try {
-            return handle(g.next());
-        } catch (ex) {
-            return $p.reject(ex);
-        }
+        return handle(g.next());
     }
 }
 
diff --git a/lib/index.js b/lib/index.js
index c6c83f7d..4db5b1f6 100644
--- a/lib/index.js
+++ b/lib/index.js
@@ -457,14 +457,6 @@ function $extend(ctx, obj) {
             }
         }
 
-        // support for generators;
-        if (cb.constructor.name === 'GeneratorFunction') {
-            taskCtx.cb = $npm.async(cb);
-            if ('txMode' in cb) {
-                taskCtx.cb.txMode = cb.txMode;
-            }
-        }
-
         var tsk = new Task(taskCtx, tag, isTX);
 
         if (taskCtx.db) {
@@ -754,9 +746,12 @@ function $task(ctx, obj, isTX) {
 
     // callback invocation helper;
     function callback() {
-        var result;
+        var result, cb = ctx.cb;
+        if (cb.constructor.name === 'GeneratorFunction') {
+            cb = $npm.async(cb);
+        }
         try {
-            result = ctx.cb.call(obj, obj); // invoking the callback function;
+            result = cb.call(obj, obj); // invoking the callback function;
         } catch (err) {
             $notify.error(ctx.options, err, {
                 client: ctx.db.client,
diff --git a/package.json b/package.json
index 4f2c7b9e..c5ff36d3 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "pg-promise",
-  "version": "2.7.1",
+  "version": "2.7.2",
   "description": "PostgreSQL via promises",
   "main": "lib/index.js",
   "scripts": {