diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml
index 84c15d8..901b921 100644
--- a/.github/workflows/docs.yaml
+++ b/.github/workflows/docs.yaml
@@ -15,7 +15,7 @@ jobs:
       - name: 🐍Setup Python
         uses: actions/setup-python@v4
         with:
-          python-version: 3.8
+          python-version: 3.10
       - name: 🔨 Setup poetry
         uses: abatilo/actions-poetry@v2.0.0
         with:
diff --git a/.github/workflows/pre-release.yaml b/.github/workflows/pre-release.yaml
index 2a0a86d..ce44575 100644
--- a/.github/workflows/pre-release.yaml
+++ b/.github/workflows/pre-release.yaml
@@ -41,7 +41,7 @@ jobs:
         uses: python-semantic-release/python-semantic-release@v8.7.0
         with:
           github_token: ${{ secrets.RELEASE_WORKFLOW }}
-          force: "prerelease"
+          prerelease: 'true'
 
       - name: 🐍 Publish package distributions to PyPI
         uses: pypa/gh-action-pypi-publish@v1.8.14
diff --git a/niceml/experiments/experimentcontext.py b/niceml/experiments/experimentcontext.py
index 87ab40e..a7c7004 100644
--- a/niceml/experiments/experimentcontext.py
+++ b/niceml/experiments/experimentcontext.py
@@ -128,11 +128,20 @@ def read_json(self, data_path: str) -> dict:
             return read_json(join(root_path, data_path), file_system=file_system)
 
     def write_image(
-        self, image: Image.Image, data_path: str, apply_last_modified: bool = True
+        self,
+        image: Image.Image,
+        data_path: str,
+        apply_last_modified: bool = True,
+        **kwargs,
     ):
         """Writes an image relative to the experiment"""
         with open_location(self.fs_config) as (file_system, root_path):
-            write_image(image, join(root_path, data_path), file_system=file_system)
+            write_image(
+                image,
+                join(root_path, data_path),
+                file_system=file_system,
+                **kwargs,
+            )
         if apply_last_modified:
             self.update_last_modified()