plotly
diff --git a/‎plotly/package_data/graphWidget.js
Lines changed: 21 additions & 4 deletions b/‎plotly/package_data/graphWidget.js
Lines changed: 21 additions & 4 deletions
diff --git a/‎plotly/widgets/graph_widget.py
Lines changed: 5 additions & 2 deletions b/‎plotly/widgets/graph_widget.py
Lines changed: 5 additions & 2 deletions
@@ -9,8 +9,10 @@
 from requests.compat import json as _json
 
 # TODO: protected imports?
-from IPython.html import widgets
-from IPython.utils.traitlets import Unicode
+# from IPython.html import widgets
+# from IPython.utils.traitlets import Unicode
+import ipywidgets as widgets
+from traitlets import Unicode
 from IPython.display import Javascript, display
 
 import plotly.plotly.plotly as py
@@ -33,6 +35,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)