diff --git a/03_Run_Locally.ipynb b/03_Run_Locally.ipynb
index fa4e09f..5c7f222 100644
--- a/03_Run_Locally.ipynb
+++ b/03_Run_Locally.ipynb
@@ -179,7 +179,7 @@
    "source": [
     "est = Estimator(source_directory=os.path.join('.', 'scripts'), \n",
     "                entry_script='TrainClassifier.py',\n",
-    "                script_params={'--data-folder': os.path.abspath('.'),\n",
+    "                script_params={'--data-folder': '/data',\n",
     "                               '--estimators': estimators,\n",
     "                               '--match': 5,\n",
     "                               '--ngrams': 2,\n",
@@ -188,8 +188,8 @@
     "                compute_target='local',\n",
     "                conda_packages=['pandas==0.23.4',\n",
     "                                'scikit-learn==0.21.3',\n",
-    "                                'lightgbm==2.2.1'],\n",
-    "                use_docker=False)"
+    "                                'lightgbm==2.2.1'])\n",
+    "est.run_config.environment.docker.arguments.extend(['-v', ''.join([os.path.abspath('.'), ':', '/data'])])"
    ]
   },
   {