diff --git a/test/prsim_plot.py b/test/prsim_plot.py
index f71283d..bb53ce8 100755
--- a/test/prsim_plot.py
+++ b/test/prsim_plot.py
@@ -36,17 +36,17 @@ Use -exclude='regex' to specify signals to exclude (or -ex).""")
 	colour_high = (98, 187, 93)
 	colour_low = (233, 115, 115)
 	for arg in argv:
-		r = re.findall(r'-c=(.+)', arg)
+		r = re.findall(r'-c=[\"\']?(.+)[\"\']?', arg)
 		if len(r) >= 1:
 			if r[0] == "ole":
 				colour_undefined = (233, 115, 115)
 				colour_high = (98, 187, 93)
 				colour_low = (90, 111, 199)
-			if r[0] == "og":
+			elif r[0] == "og":
 				colour_undefined = (255,0,0)
 				colour_high = (252, 186, 3)
 				colour_low = (20, 184, 186)
-			if r[0] == "michele":
+			elif r[0] == "michele":
 				colour_undefined = (100,100,100)
 				colour_high = (98, 187, 93)
 				colour_low = (233, 115, 115)