diff --git a/crates/rari-doc/src/html/rewriter.rs b/crates/rari-doc/src/html/rewriter.rs
index 19b5815..8b100ec 100644
--- a/crates/rari-doc/src/html/rewriter.rs
+++ b/crates/rari-doc/src/html/rewriter.rs
@@ -279,6 +279,11 @@ pub fn post_process_html<T: PageLike>(
             el.after("</figure>", ContentType::Html);
             Ok(())
         }),
+        element!("math[display=block]", |el| {
+            el.before("<figure class=\"table-container\">", ContentType::Html);
+            el.after("</figure>", ContentType::Html);
+            Ok(())
+        }),
         element!("*[data-rewriter=em]", |el| {
             el.prepend("<em>", ContentType::Html);
             el.append("</em>", ContentType::Html);