diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index e27ee41b..51a751c0 100644
--- a/.github/workflows/publish.yaml
+++ b/.github/workflows/publish.yaml
@@ -37,4 +37,4 @@ jobs:
         uses: pypa/gh-action-pypi-publish@27b31702a0e7fc50959f5ad993c78deac1bdfc29
         with:
           user: __token__
-          password: ${{ secrets.PYPI_API_TOKEN }}
+          password: ${{ secrets.TEST_PYPI_TOKEN }}
diff --git a/ai21/version.py b/ai21/version.py
index 21014090..0a3192fe 100644
--- a/ai21/version.py
+++ b/ai21/version.py
@@ -1 +1 @@
-VERSION = "2.0.0"
+VERSION = "1.9.0"
diff --git a/pyproject.toml b/pyproject.toml
index 8002b575..714af3d6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -46,7 +46,7 @@ exclude_lines = [
 
 [tool.poetry]
 name = "ai21"
-version = "2.0.0"
+version = "1.9.0"
 description = ""
 authors = ["AI21 Labs"]
 readme = "README.md"
@@ -104,11 +104,12 @@ match = "(main)"
 build_command = "pip install poetry && poetry build"
 version_source = "tag"
 commit_version_number = true
-major_on_zero = true
 commit_message = "chore(release): v{version} [skip ci]"
 
 [tool.semantic_release.branches.main]
 match = "(main)"
+prerelease_token = "rc"
+prerelease = true
 
 [tool.semantic_release.changelog.environment]
 newline_sequence = "\n"
diff --git a/setup.py b/setup.py
index 4c1f237c..8db436ba 100755
--- a/setup.py
+++ b/setup.py
@@ -1,26 +1,27 @@
 #!/usr/bin/python3
 
 import os
+import codecs
 
 from setuptools import setup, find_packages
+from ai21.version import VERSION
 
-current_folder = os.path.abspath(os.path.dirname(__file__))
-version_file_path = current_folder + "/ai21/version.py"
+readme_path = os.path.abspath(os.path.dirname(__file__))
 
-version_file = {}
-with open(version_file_path, "rt") as f:
-    exec(f.read(), version_file)
-version = version_file["__version__"]
+with codecs.open(os.path.join(readme_path, "README.md"), encoding="utf-8") as fh:
+    long_description = "\\n" + fh.read()
 
 setup(
     name="ai21",
-    version=version,
+    version=VERSION,
     license="MIT",
-    author="ai21 labs",
+    author="AI21 Labs",
     author_email="support@ai21.com",
+    long_description_content_type="text/markdown",
+    long_description=long_description,
     packages=find_packages(exclude=["tests", "tests.*"]),
+    keywords=["python", "sdk", "ai", "ai21", "jurassic", "ai21-python", "llm"],
     install_requires=[
         "requests",
     ],
-    # extras_require={"AWS": ["boto3>=1.28.82"]},
 )