diff --git a/packages/python/plotly/plotly/basewidget.py b/packages/python/plotly/plotly/basewidget.py
index 27cd8c7e99b..05df7babefd 100644
--- a/packages/python/plotly/plotly/basewidget.py
+++ b/packages/python/plotly/plotly/basewidget.py
@@ -733,12 +733,11 @@ def _handler_js2py_pointsCallback(self, change):
 
     # Display
     # -------
-    def _ipython_display_(self):
+    def _repr_mimebundle_(self, include=None, exclude=None, validate=True, **kwargs):
         """
         Handle rich display of figures in ipython contexts
         """
-        # Override BaseFigure's display to make sure we display the widget version
-        widgets.DOMWidget._ipython_display_(self)
+        widgets.DOMWidget._repr_mimebundle_(self)
 
     # Callbacks
     # ---------