diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 2043acf05074acf4f7dbc4098aec849ffe114c63..176b3705b686e8333d6dcbba13e429096fa775d2 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -8,6 +8,10 @@ Run tests:
     - docker
   image: ghcr.io/prefix-dev/pixi:latest
   allow_failure: false
+  cache:
+    key: jvm17-cache
+    paths:
+      - pythermogis/resources/JVM17/
   script:
     - export TZ=Europe/Amsterdam
     - ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
diff --git a/docs/reference/assess_optimal_chunk_size.md b/docs/reference/assess_optimal_chunk_size.md
index 4812372400d4a47490e4c8854520e5d620478027..06d0aa0292a20e0193ddfb161ffff1c01928a172 100644
--- a/docs/reference/assess_optimal_chunk_size.md
+++ b/docs/reference/assess_optimal_chunk_size.md
@@ -1,3 +1,3 @@
 # API Reference
 
-::: pythermogis.dask.assess_optimal_chunk_size
\ No newline at end of file
+::: pythermogis.dask.auto_chunk_dataset
diff --git a/mkdocs.yml b/mkdocs.yml
index 211550403caa390bcef4e8c039034b8b1a06014b..b55f50244d007a5032857274d953bf162b027978 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -63,7 +63,7 @@ nav:
       - Portfolio run and analysis: usage/portfoliorun_analysis.md
       - Customised stochastic simulations: usage/customised_stochastic_simulations.md
       - Parallelzation: usage/parallelization.md
-      - Depth optimization: usage\depth_optimization.md
+      - Depth optimization: usage/depth_optimization.md
       #- out maps and locations: usage/pvalues_map_locations.md
 
   - API Reference:
@@ -82,7 +82,7 @@ plugins:
   - mkdocstrings:
       handlers:
         python:
-          paths: ["src"]
+          paths: ["."]
           options:
             docstring_style: numpy  # or numpy
             show_source: true
diff --git a/pixi.lock b/pixi.lock
index db333a22047feb0b1a6254a8d497bed22071fedc..c41574f3160269d9211770ac29833fad23ae9e74 100644
--- a/pixi.lock
+++ b/pixi.lock
@@ -6729,8 +6729,8 @@ packages:
   timestamp: 1740946648058
 - pypi: ./
   name: pythermogis
-  version: 3.0.0
-  sha256: 09f46253d44acba8af19f5bafea6b05b35bee8022e93d0ec9cd3e2f25b73ab01
+  version: 3.0.1
+  sha256: ea171e7f4d81cfaacee0ed91c6a71bddf3a1e9b1e45d0e67383f36741d9c1560
   requires_dist:
   - jpype1>=1.5.2,<2
   - xarray==2024.9.0.*