diff --git a/setup.py b/setup.py
index e7bdc8a..5276911 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@
     install_requires=[
         "torch>=1.11",
         "pandas",
-        "polyagamma==1.3.2",
+        "polyagamma",
         "tqdm",
     ],
     extras_require={