diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index dde0303..708d0c2 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -23,15 +23,15 @@ jobs:
       - name: Build package
         run: python setup.py sdist bdist_wheel
 
-#      - name: Publish package
-#        uses: pypa/gh-action-pypi-publish@release/v1
-#        with:
-#          user: __token__
-#          password: ${{ secrets.PYPI_API_TOKEN }}
-
-      - name: Publish package to TestPyPI
+      - name: Publish package
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           user: __token__
-          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
-          repository-url: https://test.pypi.org/legacy/
+          password: ${{ secrets.PYPI_API_TOKEN }}
+
+#      - name: Publish package to TestPyPI
+#        uses: pypa/gh-action-pypi-publish@release/v1
+#        with:
+#          user: __token__
+#          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+#          repository-url: https://test.pypi.org/legacy/
diff --git a/prosto_sms/methods.py b/prosto_sms/methods.py
index 3b03a1a..0f29971 100644
--- a/prosto_sms/methods.py
+++ b/prosto_sms/methods.py
@@ -81,7 +81,7 @@ class Methods:
     def get_prices(
         self,
         *,
-        group_directions: int = None
+        group_directions: str = None
     ) -> models.GetPricesData:
         """
         Implementation of get_prices
diff --git a/prosto_sms/models.py b/prosto_sms/models.py
index 71ff90f..d50b05b 100644
--- a/prosto_sms/models.py
+++ b/prosto_sms/models.py
@@ -4,7 +4,7 @@ from pydantic import BaseModel
 
 
 class GetPricesData(BaseModel):
-    group_directions: str
+    group_directions: dict
 
 
 class GetMessageReportData(BaseModel):