diff --git a/neetbox/client/apis/_image.py b/neetbox/client/apis/_image.py
index 3fd0fc65..5ec9454b 100644
--- a/neetbox/client/apis/_image.py
+++ b/neetbox/client/apis/_image.py
@@ -169,6 +169,7 @@ def add_image(name: str, image, dataformats: str = None):
 
 
 def add_figure(
+    name,
     figure,
     close: Optional[bool] = True,
 ):
@@ -183,6 +184,6 @@ def add_figure(
         walltime: Override default walltime (time.time()) of event
     """
     if isinstance(figure, list):
-        add_image(image=figure_to_image(figure, close), dataformats="NCHW")
+        add_image(name=name, image=figure_to_image(figure, close), dataformats="NCHW")
     else:
-        add_image(image=figure_to_image(figure, close), dataformats="CHW")
+        add_image(name=name, image=figure_to_image(figure, close), dataformats="CHW")