1diff --git a/lib/index.js b/lib/index.js
2--- a/lib/index.js
3+++ b/lib/index.js
4@@ -144,6 +144,10 @@
5 const data = this.resume()
6 const node = this.stack[this.stack.length - 1]
7 assert(node.type === 'inlineMath')
8+ const source = this.sliceSerialize(token)
9+ if (source.startsWith('$$') || source.startsWith('\\[')) {
10+ node.data.hProperties.className = ['language-math', 'math-display']
11+ }
12 this.exit(token)
13 node.value = data
14 const children = /** @type {Array<HastElementContent>} */ (