diff --git a/charts/jupyter-pyspark/Chart.yaml b/charts/jupyter-pyspark/Chart.yaml
index bd5614aa..636cd0ad 100644
--- a/charts/jupyter-pyspark/Chart.yaml
+++ b/charts/jupyter-pyspark/Chart.yaml
@@ -24,7 +24,7 @@ type: application
 # This is the chart version. This version number should be incremented each time you make changes
 # to the chart and its templates, including the app version.
 # Versions are expected to follow Semantic Versioning (https://semver.org/)
-version: 3.1.0
+version: 3.1.1
 dependencies:
   - name: library-chart
     version: 2.0.3
diff --git a/charts/jupyter-pyspark/templates/statefulset.yaml b/charts/jupyter-pyspark/templates/statefulset.yaml
index 9c7b5ad7..85110b7d 100644
--- a/charts/jupyter-pyspark/templates/statefulset.yaml
+++ b/charts/jupyter-pyspark/templates/statefulset.yaml
@@ -146,8 +146,8 @@ spec:
             - -c
             - |
               echo 'initContainer make-configmaps-writable is started';
+              mkdir -p /dest/jupyter-ai
               {{ if (.Values.userPreferences.aiAssistant).enabled }}
-              mkdir /dest/jupyter-ai
               cp /src/jupyter-ai/config.json /dest/jupyter-ai/config.json
               {{- end }}
               {{- if .Values.s3.enabled }}
@@ -338,12 +338,10 @@ spec:
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
           volumeMounts:
-            - name: jupyter-local
-              mountPath: /home/{{ .Values.environment.user }}/.local/share/jupyter
-            {{ if (.Values.userPreferences.aiAssistant).enabled }}
-            - mountPath: /home/{{ .Values.environment.user }}/.local/share/jupyter/jupyter_ai
+            {{- if (.Values.userPreferences.aiAssistant).enabled }}
+            - name: config-files
+              mountPath: /home/{{ .Values.environment.user }}/.local/share/jupyter/jupyter_ai
               subPath: jupyter-ai
-              name: config-files
             {{- end }}
             - mountPath: /home/{{ .Values.environment.user }}/work
               subPath: work
diff --git a/charts/jupyter-python/Chart.yaml b/charts/jupyter-python/Chart.yaml
index 2fc8d5bf..2807d9ff 100644
--- a/charts/jupyter-python/Chart.yaml
+++ b/charts/jupyter-python/Chart.yaml
@@ -22,7 +22,7 @@ type: application
 # This is the chart version. This version number should be incremented each time you make changes
 # to the chart and its templates, including the app version.
 # Versions are expected to follow Semantic Versioning (https://semver.org/)
-version: 2.4.0
+version: 2.4.1
 dependencies:
   - name: library-chart
     version: 2.0.3
diff --git a/charts/jupyter-python/templates/statefulset.yaml b/charts/jupyter-python/templates/statefulset.yaml
index d9c34a05..5926cf8c 100644
--- a/charts/jupyter-python/templates/statefulset.yaml
+++ b/charts/jupyter-python/templates/statefulset.yaml
@@ -122,7 +122,7 @@ spec:
             - |
               echo 'initContainer make-secrets-writable is started';
               {{ if (.Values.userPreferences.aiAssistant).enabled }}
-              mkdir /dest/jupyter-ai
+              mkdir -p /dest/jupyter-ai
               cp /src/jupyter-ai/config.json /dest/jupyter-ai/config.json
               {{- end }}
               {{- if .Values.discovery.hive }}
@@ -289,12 +289,10 @@ spec:
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
           volumeMounts:
-            - name: jupyter-local
-              mountPath: /home/{{ .Values.environment.user }}/.local/share/jupyter
-            {{ if (.Values.userPreferences.aiAssistant).enabled }}
-            - mountPath: /home/{{ .Values.environment.user }}/.local/share/jupyter/jupyter_ai
-              subPath: jupyter-ai
-              name: config-files
+            {{- if (.Values.userPreferences.aiAssistant).enabled }}
+            - name: config-files
+              mountPath: /home/{{ .Values.environment.user }}/.local/share/jupyter/jupyter_ai/config.json
+              subPath: jupyter-ai/config.json
             {{- end }}
             - mountPath: /home/{{ .Values.environment.user }}/work
               subPath: work