diff --git a/layouts/shortcodes/iimg.html b/layouts/shortcodes/iimg.html
index c714d8b..63b1139 100644
--- a/layouts/shortcodes/iimg.html
+++ b/layouts/shortcodes/iimg.html
@@ -3,6 +3,7 @@
{{- $ref := .Get "ref" -}}
{{- $scale := .Get "scale" -}}
{{- $alt := or (.Get "alt") (.Get "title") -}}
+ {{- $class := .Get "class" -}}
{{- $refImg := "" -}}
{{- $refPage := "" -}}
{{- if $ref -}}
@@ -22,6 +23,6 @@
{{- else -}}
-
+
{{- end -}}