diff --git a/modules/nf-core/pyclonevi/main.nf b/modules/nf-core/pyclonevi/main.nf
index 0e63685c4626..c3fe1eebc594 100644
--- a/modules/nf-core/pyclonevi/main.nf
+++ b/modules/nf-core/pyclonevi/main.nf
@@ -24,7 +24,6 @@ process PYCLONEVI {
 
     stub:
     def prefix = task.ext.prefix ?: "${meta.id}"
-
     """
     touch ${prefix}_cluster_table.csv
     touch ${prefix}.tsv
diff --git a/modules/nf-core/pyclonevi/templates/main_script.py b/modules/nf-core/pyclonevi/templates/main_script.py
index fd17532c2722..1e88fc71aa4b 100644
--- a/modules/nf-core/pyclonevi/templates/main_script.py
+++ b/modules/nf-core/pyclonevi/templates/main_script.py
@@ -163,9 +163,7 @@ def pyclone_ctree(joint, best_fit, ctree_input):
 
     # Version
     version = (
-        subprocess.check_output("pip show pyclone-vi | grep Version | awk '{print \$NF}'", shell=True)
-        .decode()
-        .split("\\n")[0]
+        subprocess.check_output("pyclone-vi --version | sed 's/.*version //g'", shell=True).decode().split("\\n")[0]
     )
 
     f = open("versions.yml", "a")
diff --git a/modules/nf-core/pyclonevi/tests/main.nf.test b/modules/nf-core/pyclonevi/tests/main.nf.test
index d36b68c56572..2e95ca62b02e 100644
--- a/modules/nf-core/pyclonevi/tests/main.nf.test
+++ b/modules/nf-core/pyclonevi/tests/main.nf.test
@@ -28,10 +28,10 @@ nextflow_process {
             assert process.success
             assertAll(
                 { assert snapshot(
-                    sanitizeOutput(
-                        process.out,
-                        unstableKeys: ["ctree_input", "pyclone_input", "pyclone_all_fits", "pyclone_best_fit"]
-                        )).match() }
+                    sanitizeOutput(process.out,
+                        unstableKeys: ["ctree_input", "pyclone_input", "pyclone_all_fits", "pyclone_best_fit"]),
+                    path(process.out.versions_pyclonevi[0]).yaml
+                    ).match() }
             )
         }
 
diff --git a/modules/nf-core/pyclonevi/tests/main.nf.test.snap b/modules/nf-core/pyclonevi/tests/main.nf.test.snap
index 31ad42db4451..32d2e825c6f1 100644
--- a/modules/nf-core/pyclonevi/tests/main.nf.test.snap
+++ b/modules/nf-core/pyclonevi/tests/main.nf.test.snap
@@ -43,19 +43,19 @@
                     ]
                 ],
                 "versions_pyclonevi": [
-                    "versions.yml:md5,d7a72f90b5e871ad593d6b02ac78abb4"
+                    "versions.yml:md5,bc0f9ae69eb48d45a2c7389d55f7882e"
                 ]
             },
             {
                 "PYCLONEVI": {
-                    "pyclonevi": "0.1.6"
+                    "pyclonevi": "0.2.0"
                 }
             }
         ],
-        "timestamp": "2026-05-26T11:01:04.192976421",
+        "timestamp": "2026-05-26T17:22:45.350487609",
         "meta": {
             "nf-test": "0.9.5",
-            "nextflow": "26.04.0"
+            "nextflow": "26.04.2"
         }
     },
     "pyclonevi": {
@@ -105,14 +105,19 @@
                     ]
                 ],
                 "versions_pyclonevi": [
-                    "versions.yml:md5,764c065f67a1502d5158a8a018d8a388"
+                    "versions.yml:md5,3d517410bfd758e0eea9bef7f40d22e6"
                 ]
+            },
+            {
+                "PYCLONEVI": {
+                    "pyclonevi": "0.2.0"
+                }
             }
         ],
-        "timestamp": "2026-05-26T11:00:56.527463061",
+        "timestamp": "2026-05-27T07:54:19.275868186",
         "meta": {
             "nf-test": "0.9.5",
-            "nextflow": "26.04.0"
+            "nextflow": "26.04.2"
         }
     }
 }
\ No newline at end of file