diff --git a/layouts/partials/kroki/output.html b/layouts/partials/kroki/output.html
index 13499c5..cc1eb80 100644
--- a/layouts/partials/kroki/output.html
+++ b/layouts/partials/kroki/output.html
@@ -1,6 +1,7 @@
 {{- $svg := .svg }}
 {{- $output := default site.Params.kroki.output .output }}
 {{- $attrs := newScratch }}
+{{- $attrs.Set "class" "kroki" }}
 {{- range $key, $value := .attrs }}
   {{- if or
     (eq $key "type")