diff --git a/dysnix/app/tests/persistence_test.yaml b/dysnix/app/tests/persistence_test.yaml
index 00990a9f..2ee08f9e 100644
--- a/dysnix/app/tests/persistence_test.yaml
+++ b/dysnix/app/tests/persistence_test.yaml
@@ -106,8 +106,12 @@ tests:
       -
         template: statefulset.yaml
         contains:
-          path: spec.volumeClaimTemplates[0]
+          path: spec.volumeClaimTemplates
           any: true
           content:
-            name: data
-            emptyDir: {}
\ No newline at end of file
+            metadata:
+              name: data
+            spec:
+              resources:
+                requests:
+                  storage: 10Gi
diff --git a/dysnix/app/tests/statefulset-common_test.yaml b/dysnix/app/tests/statefulset-common_test.yaml
index 4d7c89cc..3bbe3c2e 100644
--- a/dysnix/app/tests/statefulset-common_test.yaml
+++ b/dysnix/app/tests/statefulset-common_test.yaml
@@ -53,6 +53,8 @@ tests:
     it: selector matchLabels
     values:
       - ./values/common.yaml
+    set:
+      app.workload.type: statefulset
     asserts:
       - isSubset:
           path: spec.selector.matchLabels
@@ -62,6 +64,8 @@ tests:
     it: statefulset labels
     values:
       - ./values/common.yaml
+    set:
+      app.workload.type: statefulset
     asserts:
      - isSubset:
         path: metadata.labels