diff --git a/dynamo/configuration.py b/dynamo/configuration.py
index 887b0f4eb..54c73b243 100755
--- a/dynamo/configuration.py
+++ b/dynamo/configuration.py
@@ -789,14 +789,17 @@ def set_figure_params(
             for more details.
     """
 
-    try:
-        import IPython
-
-        if isinstance(ipython_format, str):
-            ipython_format = [ipython_format]
-        IPython.display.set_matplotlib_formats(*ipython_format)
-    except Exception:
-        pass
+    import builtins
+
+    if getattr(builtins, "__IPYTHON__", False):
+        try:
+            import matplotlib_inline.backend_inline
+
+            if isinstance(ipython_format, str):
+                ipython_format = [ipython_format]
+            matplotlib_inline.backend_inline.set_matplotlib_formats(*ipython_format)
+        except Exception:
+            pass
 
     from matplotlib import rcParams