From e7be9d437b1690aa40abcc441c0582abfa08c7b3 Mon Sep 17 00:00:00 2001
From: nickjcroucher <n.croucher@imperial.ac.uk>
Date: Fri, 15 Nov 2024 15:27:54 +0000
Subject: [PATCH] Remove redundancy

Co-authored-by: Anmol Thapar <mr.anmolthapar@gmail.com>
---
 PopPUNK/plot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/PopPUNK/plot.py b/PopPUNK/plot.py
index 980c8576..c8e5dd6b 100644
--- a/PopPUNK/plot.py
+++ b/PopPUNK/plot.py
@@ -570,7 +570,7 @@ def outputsForCytoscape(G, G_mst, isolate_names, clustering, outPrefix, epiCsv,
       for v in G.vertices():
         comp_filter[v] = (component_assignments[v] == component_idx)
         # If using partial query graph find the component name from the clustering
-        if get_component_name and (component_assignments[v] == component_idx):
+        if get_component_name and comp_filter[v]:
           example_isolate_name = seqLabels[int(v)]
           component_name = clustering[example_cluster_title][example_isolate_name]
           get_component_name = False