diff --git a/vars/listDraftsets.groovy b/vars/listDraftsets.groovy
index b35a846..995575f 100644
--- a/vars/listDraftsets.groovy
+++ b/vars/listDraftsets.groovy
@@ -9,10 +9,12 @@
 import uk.org.gss_data.pmd_drafter.api.DraftsetsApi
 
 def call(auth) {
-    api = new DraftsetsApi()
+    def api = new DraftsetsApi()
     api.basePath = "https://production-drafter-ons-alpha.publishmydata.com"
     api.versionPath = "/v1"
     api.auth = auth
-    api.draftsetsGet(null, { result -> println(result) },
+    def ds = null
+    api.draftsetsGet(null, { result -> ds = result },
             { status, reason -> println(reason) })
+    return ds
 }
\ No newline at end of file