diff --git a/xfel_calibrate/calibrate.py b/xfel_calibrate/calibrate.py
index f8784b31b74555a9f1a67c27e094cb4f601c55bd..b8ecfb7ae3dbe8e538618ff6e94be172dbe86100 100755
--- a/xfel_calibrate/calibrate.py
+++ b/xfel_calibrate/calibrate.py
@@ -369,7 +369,7 @@ def flatten_list(l):
     return "_".join([str(flatten_list(v)) for v in l]) if isinstance(l, list) else l
 
 
-def set_figure_format(nb, param):
+def set_figure_format(nb, enable_vector_format):
     """
     Set svg format in inline backend for figures
 
@@ -381,7 +381,7 @@ def set_figure_format(nb, param):
     :param param: value of corresponding parameter
     """
 
-    if param:
+    if enable_vector_format:
         cell = get_cell_n(nb, 'code', 1)
         cell.source += "\n%config InlineBackend.figure_formats = ['svg']\n"