diff --git a/environment.yml b/environment.yml
index 9dc036b..bc88f5a 100644
--- a/environment.yml
+++ b/environment.yml
@@ -8,6 +8,7 @@ dependencies:
   - numpy
   - pandas
   - plotly
+  - chart-studio
   - requests
   - seaborn
   - traitlets
diff --git a/python/plotly.ipynb b/python/plotly.ipynb
index 003c61f..0af2acd 100644
--- a/python/plotly.ipynb
+++ b/python/plotly.ipynb
@@ -169,7 +169,7 @@
    },
    "outputs": [],
    "source": [
-    "import plotly.plotly as py\n",
+    "import chart_studio.plotly as py\n",
     "import plotly.graph_objs as go\n",
     "\n",
     "import pandas as pd\n",