diff --git a/datamol/viz/_viz.py b/datamol/viz/_viz.py
index 91a88216..b340d1f5 100644
--- a/datamol/viz/_viz.py
+++ b/datamol/viz/_viz.py
@@ -129,6 +129,7 @@ def to_image(
         highlightAtomLists=_highlight_atom,
         highlightBondLists=_highlight_bond,
         drawOptions=draw_options,
+        maxMols=max_mols,
         **_kwargs,
     )