diff --git a/packages/python/plotly/codegen/__init__.py b/packages/python/plotly/codegen/__init__.py
index 6676744d2a4..a50bacf0b38 100644
--- a/packages/python/plotly/codegen/__init__.py
+++ b/packages/python/plotly/codegen/__init__.py
@@ -259,9 +259,9 @@ def perform_codegen():
     from distutils.version import LooseVersion
     if LooseVersion(ipywidgets.__version__) >= LooseVersion('7.0.0'):
         from ._figurewidget import FigureWidget
+        __all__.append("FigureWidget")
     del LooseVersion
     del ipywidgets
-    __all__.append("FigureWidget")
 except ImportError:
     pass
 """
diff --git a/packages/python/plotly/plotly/graph_objs/__init__.py b/packages/python/plotly/plotly/graph_objs/__init__.py
index ee0cde75f81..0e99a0849b6 100644
--- a/packages/python/plotly/plotly/graph_objs/__init__.py
+++ b/packages/python/plotly/plotly/graph_objs/__init__.py
@@ -100182,8 +100182,9 @@ def __init__(
 
     if LooseVersion(ipywidgets.__version__) >= LooseVersion("7.0.0"):
         from ._figurewidget import FigureWidget
+
+        __all__.append("FigureWidget")
     del LooseVersion
     del ipywidgets
-    __all__.append("FigureWidget")
 except ImportError:
     pass