diff --git a/run/create_plots.py b/run/create_plots.py
index fda5204fdf34b3eccccfe75527fd709b3407f11c..c2d7d544e7d27f47cee8f2ff63df1c0c5d2fceff 100644
--- a/run/create_plots.py
+++ b/run/create_plots.py
@@ -48,7 +48,7 @@ def read_data(file, frame):
         network = "_".join(model_name.split("_")[1:])
         method = words[1] if len(words) > 2 else words[0]
 
-        if "vnnclass1" in model_name:
+        if "vnnclass1" or "vnnclass3" in model_name:
             method=words[-3]
 
         if model_name in ["tanet_car_2", "tanet_car_3"]: