diff --git a/baselines/fedvssl/fedvssl/finetune_preprocess.py b/baselines/fedvssl/fedvssl/finetune_preprocess.py
index e0a34c8f60d2..1f40c2b79a3e 100644
--- a/baselines/fedvssl/fedvssl/finetune_preprocess.py
+++ b/baselines/fedvssl/fedvssl/finetune_preprocess.py
@@ -46,9 +46,10 @@ def args_parser():
 try:
     params = params["arr_0"].item()
 except:
+    # For the cases where the weights are stored as NumPy arrays instead of parameters
     params = [
         np.array(v) for v in list(params["arr_0"])
-    ]  # for the cases where the weights are stored as NumPy arrays instead of parameters
+    ]
 try:
     params = parameters_to_ndarrays(params)
 except: