diff --git a/.github/workflows/build-docs.yaml b/.github/workflows/build-docs.yaml
index 9e290368..128f8171 100644
--- a/.github/workflows/build-docs.yaml
+++ b/.github/workflows/build-docs.yaml
@@ -18,7 +18,7 @@ jobs:
           repository: Nixtla/docs
           ref: scripts
           path: docs-scripts
-      - uses: actions/setup-python@39cd14951b08e74b54015e9e001cdefcf80e669f # 5.1.1
+      - uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # 5.2.0
         with:
           python-version: "3.10"
       - name: Build docs
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index e9946990..dba53a10 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -27,7 +27,7 @@ jobs:
         uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
 
       - name: Set up environment
-        uses: actions/setup-python@39cd14951b08e74b54015e9e001cdefcf80e669f # 5.1.1
+        uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # 5.2.0
         with:
           python-version: ${{ matrix.python-version }}
 
@@ -50,7 +50,7 @@ jobs:
         uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
 
       - name: Set up environment
-        uses: actions/setup-python@39cd14951b08e74b54015e9e001cdefcf80e669f # 5.1.1
+        uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # 5.2.0
         with:
           python-version: ${{ matrix.python-version }}
 
@@ -71,7 +71,7 @@ jobs:
         uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
 
       - name: Set up python
-        uses: actions/setup-python@39cd14951b08e74b54015e9e001cdefcf80e669f # 5.1.1
+        uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # 5.2.0
         with:
           python-version: "3.10"
 
@@ -86,7 +86,7 @@ jobs:
     steps:
       - uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
 
-      - uses: actions/setup-python@39cd14951b08e74b54015e9e001cdefcf80e669f # 5.1.1
+      - uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # 5.2.0
         with:
           python-version: "3.12"
 
@@ -104,7 +104,7 @@ jobs:
     steps:
       - uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
 
-      - uses: actions/setup-python@39cd14951b08e74b54015e9e001cdefcf80e669f # 5.1.1
+      - uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # 5.2.0
         with:
           python-version: "3.10"
 
diff --git a/.github/workflows/lint.yaml b/.github/workflows/lint.yaml
index c1c8373c..5d45c136 100644
--- a/.github/workflows/lint.yaml
+++ b/.github/workflows/lint.yaml
@@ -14,7 +14,7 @@ jobs:
         uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
 
       - name: Set up python
-        uses: actions/setup-python@39cd14951b08e74b54015e9e001cdefcf80e669f # 5.1.1
+        uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # 5.2.0
         with:
           python-version: '3.10'
           cache: 'pip'
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 4e8910c6..a6bccd45 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -20,7 +20,7 @@ jobs:
         uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
 
       - name: Set up python
-        uses: actions/setup-python@39cd14951b08e74b54015e9e001cdefcf80e669f # 5.1.1
+        uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # 5.2.0
         with:
           python-version: '3.10' 
 
@@ -31,4 +31,4 @@ jobs:
         run: python -m build -sw
 
       - name: Publish package to PyPI
-        uses: pypa/gh-action-pypi-publish@ec4db0b4ddc65acdf4bff5fa45ac92d78b56bdf0 # v1.9.0
+        uses: pypa/gh-action-pypi-publish@8a08d616893759ef8e1aa1f2785787c0b97e20d6 # v1.10.0
diff --git a/nbs/docs/getting-started/quick_start_distributed.ipynb b/nbs/docs/getting-started/quick_start_distributed.ipynb
index 96431499..87e02a05 100644
--- a/nbs/docs/getting-started/quick_start_distributed.ipynb
+++ b/nbs/docs/getting-started/quick_start_distributed.ipynb
@@ -51,7 +51,7 @@
     "import git\n",
     "import numpy as np\n",
     "import pandas as pd\n",
-    "from botocore.exceptions import NoCredentialsError\n",
+    "import s3fs\n",
     "from sklearn.base import BaseEstimator\n",
     "\n",
     "from mlforecast.distributed import DistributedMLForecast\n",
@@ -563,8 +563,10 @@
     "save_path = f's3://nixtla-tmp/mlf/{save_dir}'\n",
     "tmpdir = tempfile.TemporaryDirectory()\n",
     "try:\n",
+    "    s3fs.S3FileSystem().ls('s3://nixtla-tmp/')\n",
     "    fcst.save(save_path)\n",
-    "except NoCredentialsError:\n",
+    "except Exception as e:\n",
+    "    print(e)\n",
     "    save_path = f'{tmpdir.name}/{save_dir}'\n",
     "    fcst.save(save_path)"
    ]
@@ -1249,8 +1251,10 @@
     "save_dir = build_unique_name('spark')\n",
     "save_path = f's3://nixtla-tmp/mlf/{save_dir}'\n",
     "try:\n",
+    "    s3fs.S3FileSystem().ls('s3://nixtla-tmp/')\n",
     "    fcst.save(save_path)\n",
-    "except NoCredentialsError:\n",
+    "except Exception as e:\n",
+    "    print(e)\n",
     "    save_path = f'{tmpdir.name}/{save_dir}'\n",
     "    fcst.save(save_path)"
    ]
@@ -1805,8 +1809,10 @@
     "save_dir = build_unique_name('ray')\n",
     "save_path = f's3://nixtla-tmp/mlf/{save_dir}'\n",
     "try:\n",
+    "    s3fs.S3FileSystem().ls('s3://nixtla-tmp/')\n",
     "    fcst.save(save_path)\n",
-    "except NoCredentialsError:\n",
+    "except Exception as e:\n",
+    "    print(e)\n",
     "    save_path = f'{tmpdir.name}/{save_dir}'\n",
     "    fcst.save(save_path)"
    ]