1 files changed, 1 insertions, 1 deletions
diff --git a/gui.py b/gui.py
index fdc587b..f09e9df 100644
--- a/gui.py
+++ b/gui.py
@@ -205,7 +205,7 @@ def draw_frame():
     cells = sorted(cells, key = lambda x: x.mass)
     
     viruses = list(filter(lambda x: x.is_virus, c.world.cells.values()))
-    viruses = sorted(cells, key = lambda x: x.mass)
+    viruses = sorted(viruses, key = lambda x: x.mass)
     
     for cell in food:
         draw_cell(cell)