diff --git a/polyapi/function_cli.py b/polyapi/function_cli.py
index b888d08..bff9a90 100644
--- a/polyapi/function_cli.py
+++ b/polyapi/function_cli.py
@@ -91,8 +91,7 @@ def function_add_or_update(
         function_id = resp.json()["id"]
         print(f"Function ID: {function_id}")
         if generate:
-            contexts=generate_contexts.split(",") if generate_contexts else None
-            generate_library(contexts=contexts)
+            generate_library()
     else:
         print("Error adding function.")
         print(resp.status_code)
diff --git a/pyproject.toml b/pyproject.toml
index 6a114d2..6881a0d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@ requires = ["setuptools>=61.2", "wheel"]
 
 [project]
 name = "polyapi-python"
-version = "0.3.6.dev0"
+version = "0.3.7.dev0"
 description = "The Python Client for PolyAPI, the IPaaS by Developers for Developers"
 authors = [{ name = "Dan Fellin", email = "dan@polyapi.io" }]
 dependencies = [