diff --git a/src/methods/geneformer/config.vsh.yaml b/src/methods/geneformer/config.vsh.yaml
index 6be7998c..a98ab008 100644
--- a/src/methods/geneformer/config.vsh.yaml
+++ b/src/methods/geneformer/config.vsh.yaml
@@ -21,20 +21,24 @@ info:
   method_types: [embedding]
   variants:
     geneformer_12L_95M_i4096:
-      model: "gf-12L-95M-i4096"
+      model: gf-12L-95M-i4096
     geneformer_6L_30M_i2048:
-      model: "gf-6L-30M-i2048"
+      model: gf-6L-30M-i2048
     geneformer_12L_30M_i2048:
-      model: "gf-12L-30M-i2048"
+      model: gf-12L-30M-i2048
     geneformer_20L_95M_i4096:
-      model: "gf-20L-95M-i4096"
+      model: gf-20L-95M-i4096
 
 arguments:
-  - name: "--model"
-    type: "string"
+  - name: --model
+    type: string
     description: String representing the Geneformer model to use
-    choices: ["gf-6L-30M-i2048", "gf-12L-30M-i2048", "gf-12L-95M-i4096", "gf-20L-95M-i4096"]
-    default: "gf-12L-95M-i4096"
+    choices:
+      - gf-6L-30M-i2048
+      - gf-12L-30M-i2048
+      - gf-12L-95M-i4096
+      - gf-20L-95M-i4096
+    default: gf-12L-95M-i4096
 
 resources:
   - type: python_script
@@ -48,9 +52,9 @@ engines:
     setup:
       - type: python
         pip:
-        - pyarrow<15.0.0a0,>=14.0.1
-        - huggingface_hub
-        - git+https://huggingface.co/ctheodoris/Geneformer.git
+          - pyarrow<15.0.0a0,>=14.0.1
+          - huggingface_hub
+          - git+https://huggingface.co/ctheodoris/Geneformer.git
 
 runners:
   - type: executable