diff --git a/scripts/compareStationListVLBI.py b/scripts/compareStationListVLBI.py
index 8244addc93572220d03adec38a4fdbd4c7dafbd6..9333d702fa1435c182809b70c8957f5dab103754 100644
--- a/scripts/compareStationListVLBI.py
+++ b/scripts/compareStationListVLBI.py
@@ -36,7 +36,6 @@ def plugin_main(args, **kwargs):
     else:
         MS = mslist[0]
         pass
-
     ## reading ANTENNA table of MS
     logging.info('Collecting information from the ANTENNA table.')
     antennaTable = pt.table(MS + "::ANTENNA", ack = False)
diff --git a/steps/check_station_mismatch.cwl b/steps/check_station_mismatch.cwl
index 31396726d6c83780618137a18d8d3f611d46725e..2869f38f42412c7b3906c20a2aa2f089892fec5f 100644
--- a/steps/check_station_mismatch.cwl
+++ b/steps/check_station_mismatch.cwl
@@ -33,14 +33,19 @@ requirements:
           entry: |
             import sys
             import json
+            import yaml
             import os
             from compareStationListVLBI import plugin_main as compareStationList
 
             mss = sys.argv[1:]
-            inputs = json.loads(r"""$(inputs)""")
+            try:
+                inputs = json.loads(r"""$(inputs)""")
+            except:
+                inputs = yaml.loads(r"""$(inputs)""")
             h5parmdb = inputs['solset']['path']
             solset_name = inputs['solset_name']
             filter = inputs['filter_baselines']
+            print(mss)
 
             output = compareStationList(mss, 
                                         h5parmdb = h5parmdb, 
diff --git a/test.yaml b/test.yaml
new file mode 100644
index 0000000000000000000000000000000000000000..01b5af99b861c229450265e3f1258d8b72f9b584
--- /dev/null
+++ b/test.yaml
@@ -0,0 +1,20 @@
+---
+msin:
+    - class: "Directory"
+      path : "/home/tjlv53/software/test_data/L693719_SB000_uv-compressed.MS"
+    - class: "Directory"
+      path : "/home/tjlv53/software/test_data/L693719_SB001_uv-compressed.MS"
+    - class: "Directory"
+      path : "/home/tjlv53/software/test_data/L693719_SB002_uv-compressed.MS"
+    - class: "Directory"
+      path : "/home/tjlv53/software/test_data/L693719_SB003_uv-compressed.MS"
+    - class: "Directory"
+      path : "/home/tjlv53/software/test_data/L693719_SB004_uv-compressed.MS"
+    - class: "Directory"
+      path : "/home/tjlv53/software/test_data/L693719_SB005_uv-compressed.MS"
+solset:
+    class: "File"
+    path : "/home/tjlv53/software/test_data/Pre-Facet-Target/results/cal_values/solutions.h5"
+parset:
+    class: "File"
+    path: "/home/tjlv53/software/VLBI-cwl/dp3.parset"