plotly
diff --git a/‎plotly/widgets/graphWidget.js
Lines changed: 17 additions & 8 deletions b/‎plotly/widgets/graphWidget.js
Lines changed: 17 additions & 8 deletions
diff --git a/‎plotly/widgets/graph_widget.py
Lines changed: 11 additions & 2 deletions b/‎plotly/widgets/graph_widget.py
Lines changed: 11 additions & 2 deletions
@@ -9,8 +9,16 @@
 
 
 # TODO: protected imports?
-from IPython.html import widgets
-from IPython.utils.traitlets import Unicode
+try:
+    import ipywidgets as widgets
+except ImportError:
+    from IPython.html import widgets
+
+try:
+    from traitlets import Unicode
+except ImportError:
+    from IPython.utils.traitlets import Unicode
+
 from IPython.display import Javascript, display
 
 import plotly.plotly.plotly as py
@@ -33,6 +41,7 @@ class GraphWidget(widgets.DOMWidget):
     Notebooks.
     """
     _view_name = Unicode('GraphView', sync=True)
+    _view_module = Unicode('GraphViewModule', sync=True)
     _message = Unicode(sync=True)
     _graph_url = Unicode(sync=True)
     _new_url = Unicode(sync=True)