diff --git a/src/MdToHTML.hs b/src/MdToHTML.hs
index 1c4bba4..4695e32 100644
--- a/src/MdToHTML.hs
+++ b/src/MdToHTML.hs
@@ -251,15 +251,16 @@ parseImage :: Parser MdToken
parseImage = do
char '!'
link <- parseLink
+ cssClasses <- optional $ try parseCssClasses
case link of
- Link text path -> return $ Image text path
+ Link text path -> return $ Image text path cssClasses
_ -> empty -- This should never be reached
parseFigure = do
img <- parseImage
void (string doubleNewlineText) <|> eof
case img of
- Image text path -> return $ Figure text path
+ Image text path cssClasses -> return $ Figure text path cssClasses
_ -> return img
-- Parse an escaped character