diff --git a/machine-learning.ipynb b/machine-learning.ipynb
index 31d43cd..0f18685 100644
--- a/machine-learning.ipynb
+++ b/machine-learning.ipynb
@@ -57,7 +57,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import dask_ml.joblib  # register the distriubted backend\n",
     "from sklearn.datasets import make_classification\n",
     "from sklearn.svm import SVC\n",
     "from sklearn.model_selection import GridSearchCV\n",
@@ -125,7 +124,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from sklearn.externals import joblib\n",
+    "import joblib\n",
     "\n",
     "with joblib.parallel_backend('dask', scatter=[X, y]):\n",
     "    grid_search.fit(X, y)"