diff --git a/gemmbench/gemm_bench.py b/gemmbench/gemm_bench.py
index 016c18b..f1af4d6 100644
--- a/gemmbench/gemm_bench.py
+++ b/gemmbench/gemm_bench.py
@@ -73,7 +73,7 @@ def compile_gemm(tag, config, kernel_dir, vmfb_dir, target, extra_compiler_args,
 
     if args.roofline:
         for dtype in requested_dtypes:
-            roofline(args.roofline, f"{args.plot.split(".")[0]}_{dtype}.png", args.batch, dtype, args.model)
+            roofline(args.roofline, f"{args.plot.split('.')[0]}_{dtype}.png", args.batch, dtype, args.model)
         sys.exit()
 
     tk = args.tk