diff --git a/packages/kestrel_jupyter/src/kestrel_jupyter_kernel/display.py b/packages/kestrel_jupyter/src/kestrel_jupyter_kernel/display.py
index 321cf9b8..2c7ba38f 100644
--- a/packages/kestrel_jupyter/src/kestrel_jupyter_kernel/display.py
+++ b/packages/kestrel_jupyter/src/kestrel_jupyter_kernel/display.py
@@ -54,9 +54,9 @@ def to_html_blocks(d: Display) -> Iterable[str]:
)
fig_buffer = BytesIO()
plt.savefig(fig_buffer, format="png")
- img = data_uri = base64.b64encode(fig_buffer.getvalue()).decode("utf-8")
- imgx = f''
- yield imgx
+ img_base64 = base64.b64encode(fig_buffer.getvalue()).decode("utf-8")
+ img_tag = f''
+ yield img_tag
if isinstance(graphlet.action, NativeQuery):
native_query = graphlet.action