diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index c8e75c96..831b3373 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -14,7 +14,7 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        poetry-version: [1.1.6]
+        poetry-version: [1.1.8]
     if: ${{ github.event_name == 'pull_request' }}
     steps:
       #----------------------------------------------
diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index 0700ad0e..58cfbe87 100644
--- a/.github/workflows/publish.yaml
+++ b/.github/workflows/publish.yaml
@@ -10,7 +10,7 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        poetry-version: [1.1.6]
+        poetry-version: [1.1.8]
     steps:
       #----------------------------------------------
       #       check-out repo and set-up python
diff --git a/pyproject.toml b/pyproject.toml
index 998284b8..9dd805c3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pgmax"
-version = "0.1.0"
+version = "0.2.0"
 description = "Loopy belief propagation for factor graphs on discrete variables, in JAX!"
 authors = ["Nishanth Kumar <nkumar@vicarious.com>", "Stannis Zhou <stannis@vicarious.com>", "Wolfgang Lehrach <wolf@vicarious.com>", "Miguel Lazaro-Gredilla <miguel@vicarious.com>"]
 # Additional info