diff --git a/charts/vscode-pyspark/Chart.yaml b/charts/vscode-pyspark/Chart.yaml
index d5fec8c5..f78e37ec 100644
--- a/charts/vscode-pyspark/Chart.yaml
+++ b/charts/vscode-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: 2.2.10
+version: 2.2.11
 dependencies:
   - name: library-chart
     version: 1.6.13
diff --git a/charts/vscode-pyspark/templates/statefulset.yaml b/charts/vscode-pyspark/templates/statefulset.yaml
index be31cca3..fb122906 100644
--- a/charts/vscode-pyspark/templates/statefulset.yaml
+++ b/charts/vscode-pyspark/templates/statefulset.yaml
@@ -166,10 +166,9 @@ spec:
               awk 'BEGIN {c=0;} /BEGIN CERT/{c++} { print > "/dest/cacerts/cert." c ".crt"}' < /tmp/ca.pem;
               {{- end }}
           volumeMounts:
-            {{ if (.Values.userPreferences.aiAssistant).enabled }}
-            - mountPath: /home/{{ .Values.environment.user }}/.continue
-              subPath: continue
-              name: config-files
+            {{- if (.Values.userPreferences.aiAssistant).enabled }}
+            - name: secret-assistant
+              mountPath: /src/continue
             {{- end }}
             - name: config-files
               mountPath: /dest
@@ -318,6 +317,11 @@ spec:
               name: home
             - mountPath: /dev/shm
               name: dshm
+            {{ if (.Values.userPreferences.aiAssistant).enabled }}
+            - mountPath: /home/{{ .Values.environment.user }}/.continue
+              subPath: continue
+              name: config-files
+            {{- end }}
             {{- if (.Values.s3).enabled }}
             - mountPath: /opt/hadoop/etc/hadoop/core-site.xml
               subPath: coresite/core-site.xml
diff --git a/charts/vscode-python/Chart.yaml b/charts/vscode-python/Chart.yaml
index 56963c35..84c50fb7 100644
--- a/charts/vscode-python/Chart.yaml
+++ b/charts/vscode-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.2.10
+version: 2.2.11
 dependencies:
   - name: library-chart
     version: 1.6.13
diff --git a/charts/vscode-python/templates/statefulset.yaml b/charts/vscode-python/templates/statefulset.yaml
index 7441a91a..00b2331d 100644
--- a/charts/vscode-python/templates/statefulset.yaml
+++ b/charts/vscode-python/templates/statefulset.yaml
@@ -130,10 +130,9 @@ spec:
               awk 'BEGIN {c=0;} /BEGIN CERT/{c++} { print > "/dest/cacerts/cert." c ".crt"}' < /tmp/ca.pem;
               {{- end }}
           volumeMounts:
-            {{ if (.Values.userPreferences.aiAssistant).enabled }}
-            - mountPath: /home/{{ .Values.environment.user }}/.continue
-              subPath: continue
-              name: config-files
+            {{- if (.Values.userPreferences.aiAssistant).enabled }}
+            - name: secret-assistant
+              mountPath: /src/continue
             {{- end }}
             - name: config-files
               mountPath: /dest
@@ -258,6 +257,11 @@ spec:
               name: home
             - mountPath: /dev/shm
               name: dshm
+            {{ if (.Values.userPreferences.aiAssistant).enabled }}
+            - mountPath: /home/{{ .Values.environment.user }}/.continue
+              subPath: continue
+              name: config-files
+            {{- end }}
             {{- if (.Values.discovery).hive }}
             - name: config-files
               mountPath: /opt/hive/conf/hive-site.xml